search

evandez / neuron-descriptions

Natural Language Descriptions of Deep Visual Features, ICLR 2022
http://milan.csail.mit.edu
MIT License
62 stars 7 forks source link

'unkown milannotation set' error for custom model and dataset #18

Open bonfab opened 2 years ago

bonfab commented 2 years ago

Trying to run the neuron-descriptions/scripts/compute_milan_descriptions.py with a custom model and dataset. I already calculated the exemplars, but now when I run

python3 "neuron-descriptions/scripts/compute_milan_descriptions.py" resnet18_pcs imagenet_custom --device cuda

I receive KeyError: 'unknown milannotations set: resnet18_pcs/imagenet_custom'

How do I compute the descriptions for custom model and dataset?

evandez commented 1 year ago

Sorry, I need to add some documents on this. You need to add an entry for your model and dataset here: https://github.com/evandez/neuron-descriptions/blob/main/src/milannotations/loaders.py