fit to fitting_stimuli in LabelBehavior instead of assuming pretrained imagenet

brain-score / model-tools

Helper functions to extract model activations and translate from Machine Learning to Neuroscience

MIT License

8 stars 27 forks source link

fit to fitting_stimuli in LabelBehavior instead of assuming pretrained imagenet #50

Open mschrimpf opened 3 years ago

mschrimpf commented 3 years ago

specifically fit classifier for label output.

resolve https://github.com/brain-score/model-tools/issues/49 merge after https://github.com/brain-score/brain-score/pull/263

remaining problem: extracting activations for ImageNet takes over 4 hours, even on GPU. We have yet to implement efficient batching and pre-loading of images.