CSAILVision / NetDissect-Lite

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

How to use Densenet model ? #6

Closed snorlaxse closed 5 years ago

snorlaxse commented 5 years ago

I modify the settings.py (MODEL = 'densenet161') and download the whole_densenet161_places365_python36.pth.tar. And run 'main.py' import errors as below:

AttributeError: 'BatchNorm2d' object has no attribute 'track_running_stats'
snorlaxse commented 5 years ago

sorry, the issuse same as #3

pipixia-bit commented 2 years ago

hello, where can you find the whole_densenet161_places365_python36.pth.tar