train.py 问题

[wzjialang]

我按照README中的操作步骤运行 python train.py --scale_factor 4 --device cuda:0 --batch_size 32 --n_epochs 80 --n_steps 30 系统给出如下错误：

Traceback (most recent call last):
File "train.py", line 107, in <module>
main(cfg)
File "train.py", line 103, in main
train(train_loader, cfg)
File "train.py", line 80, in train
psnr_epoch.append(cal_psnr(HR_left[:,:,:,64:].data.cpu(), SR_left[:,:,:,64:].data.cpu()))
File "/home/xujialang/PASSRnet/utils.py", line 84, in cal_psnr
return measure.compare_psnr(img1_np, img2_np)
File "/home/xujialang/anaconda3/envs/passrnet/lib/python3.6/site-packages/skimage/measure/simple_metrics.py", line 65, in compare_psnr
return peak_signal_noise_ratio(im_true, im_test, data_range=data_range)
File "/home/xujialang/anaconda3/envs/passrnet/lib/python3.6/site-packages/skimage/metrics/simple_metrics.py", line 149, in peak_signal_noise_ratio
"im_true has intensity values outside the range expected for "
ValueError: im_true has intensity values outside the range expected for its data type.  Please manually specify the data_range

我将'cuda:0' 切换为 'cuda:1' 或者其他GPU，错误就消失了，请问这是什么原因呢？