Caffe import is unnecessary and not used

CSAILVision / NetDissect

Network Dissection http://netdissect.csail.mit.edu for quantifying interpretability of deep CNNs.

MIT License

449 stars 115 forks source link

Caffe import is unnecessary and not used #9

Closed mlosch closed 6 years ago

mlosch commented 6 years ago

https://github.com/CSAILVision/NetDissect/blob/6c6fc51a45b6d6bfc78b83c9b372ae04cb94ffc6/src/netprobe_pytorch.py#L19

Line 19-21 and 32-33 can be removed without harming the functionality.

zhoubolei commented 6 years ago

Thanks for reporting the issue. I fixed it. The pytorch version of netdissect is still in beta test. Please let us know if there is other issue. Bolei