search

VITA-Group / EnlightenGAN

[IEEE TIP] "EnlightenGAN: Deep Light Enhancement without Paired Supervision" by Yifan Jiang, Xinyu Gong, Ding Liu, Yu Cheng, Chen Fang, Xiaohui Shen, Jianchao Yang, Pan Zhou, Zhangyang Wang
Other
890 stars 198 forks source link

Update to newest torch version? #31

Closed guantinglin closed 4 years ago

guantinglin commented 4 years ago

Hi,

Thank for the great work.

One short question, why did you implement this repo with old version or torch(0.3) instead of new torch(>1.0)? Is there any advantage using old version of torch?

BTW, will you plan to upgrade the code to support newest version of torch?

yifanjiang19 commented 4 years ago

This work is done in last year while new version may not have been released. If you need to use new version, just replace all .data[0] with .item()