BrainModel interface for CORnet-Z

[arjung128]

Hi! First off, thanks for making all of this amazing work open source!

I have a model which is a variant of the CORnet-Z model, but I am unsure how to make it an instance of the BrainModel interface. I took a look at the Jupyter notebook you provided, but I don't fully understand the various functions, and I'm not sure what these functions should look like for my case.

I was wondering -- would it be possible for you to provide an example of the BrainModel interface for the CORnet-Z model? That would make it a lot easier for me to understand the functions and make the necessary modifications to my model. Thanks!

[arjung128]

Related question: Would CORnet-Z be mapped correctly if it were submitted as a base model (as opposed to a brain model) with the layers: ['V1', 'V2', 'V4', 'IT', 'decoder.avgpool', 'decoder.output']?