search

CSAILVision / NetDissect-Lite

Light version of Network Dissection for Quantifying Interpretability of Networks
215 stars 51 forks source link

Hello All, #14

Open KarthikeyanG44 opened 5 years ago

KarthikeyanG44 commented 5 years ago

Hello, I have been trying to run the net-dissect on a ubuntu 16.04 through SSH. The main.py file throws up a StopIteration error after extracting the features. I tried going through the code and it seems to show up from the segmentation prefetcher class. I am not entirely sure how this works. I am attaching a screenshot of the error. Any help is greatly appreciated. Error_1

PS : Followed every instruction given in the description of this page

jayelm commented 4 years ago

Replace raise StopIteration with return where it appears. You're probably on python 3.7 where generator semantics has changed https://www.python.org/dev/peps/pep-0479/