预测时只能调用一张显卡
19971483525 opened this issue · comments
您好，我想问题一下这个代码在预测的时候，好像只能调用一张显卡，这个问题，能解决不，预测的时候调用多张显卡
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