CSAILVision / gandissect

Pytorch-based tools for visualizing and understanding the neurons of a GAN. https://gandissect.csail.mit.edu/
MIT License
1.77k stars 283 forks source link

Support for pytorch 1.0.1 #9

Closed Richardwan7 closed 5 years ago

Richardwan7 commented 5 years ago

Will there be plan to have the script support pytorch 1.0.1? I got the following error when executing ./travis.sh with pytorch 1.0.1: "ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead"

davidbau commented 5 years ago

Yup, we'll move to support pytorch 1.0.1. The semantic image segmentation that we use currently depends on pytorch 0.4 but they plan to move to pytorch 1, so when that is ready, we'll upgrade also.

davidbau commented 5 years ago

This should be fixed now.