search

tamarott / SinGAN

Official pytorch implementation of the paper: "SinGAN: Learning a Generative Model from a Single Natural Image"
https://tamarott.github.io/SinGAN.htm
Other
3.31k stars 611 forks source link

what are the working pytorch and torchvision version #101

Closed jxcodetw closed 2 years ago

jxcodetw commented 4 years ago

I tried to run the training in Google Colab with torch==1.5 and torchvision==0.6.0+cu101 But after some testing I found that: It can only work with torch==1.4.0 and torchvision==0.5

Should we migrate the code to newer version or fix the version in requirements.txt?

tamarott commented 4 years ago

Please work with torch==1.4.0 and torchvision==0.5

DonghweeYoon commented 4 years ago

Why not notice this training environment in README? My friend had a hard time to handle the version issue.