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

erro on compute_milan_descriptions.py #12

Closed Fatemeh490 closed 2 years ago

Fatemeh490 commented 2 years ago

I faced an error when computing Milan descriptions with running code below: ! python3 -m scripts.compute_milan_descriptions biggan imagenet --device cuda

the error is located on line 71 of compute_milan_descriptions.py. with results_csv_file.open('r') as handle: I think it should be: with results_csv_file.open('w') as handle:

evandez commented 2 years ago

Good catch, fixed!