search

facebookresearch / FaderNetworks

Fader Networks: Manipulating Images by Sliding Attributes - NIPS 2017
Other
760 stars 109 forks source link

pytorch version? #16

Open S-Ellis opened 4 years ago

S-Ellis commented 4 years ago

Which version of pytorch is required to run the code? When I try 0.3.0 I get this attribute error trying to run the interpolate example with downloaded models:

AttributeError: Can't get attribute '_rebuild_tensor_v2' on <module 'torch._utils'>

When I try 0.4 I get a different attribute error:

AttributeError: 'BatchNorm2d' object has no attribute 'track_running_stats'

Presumably it's some version inbetween but I can't work it out.

Thanks

qzq2514 commented 4 years ago

hello @S-Ellis ,I had the same problem as you. Have you solved it?

S-Ellis commented 4 years ago

@qzq2514 I couldn't find a fix. In the end I can only assume that the pre-trained models were created with a different version of pytorch than the released code uses. I had to re-train to get any models at all, though the models produced by re-training were quite useless (awful quality images, although changing attributes did have some effect).

I recommend either trying your luck retraining from scratch with their code, or (and this would be more reliable), re-implementing from scratch yourself on whichever pytorch version you normally use.

Good luck!