lhjthp/HNCT

训练结果差距悬殊

Opened this issue · 5 comments

1AnXu commented

image
按照论文进行训练,达不到/test/x2/x2.pt的效果(在DIV2K验证集上PSNR=34.743),为什么我的只达到了PSNR=34.530

1AnXu commented

我使用的命令如下python main.py --scale 2 --patch_size 64 --save hnct_x2_b16 --reset --model hnct --n_GPUs 2 --device_ids "0,1" --batch_size 16 --test_every 1000 ,作者能提供您们的训练命令吗

@1AnXu 请问你训练用了多久?

@1AnXu 请问你训练用了多久?

3~4 days

@1AnXu 麻烦请问你使用的pytorch及cuda是那个版本,及显卡配置?

@1AnXu 麻烦请问你使用的pytorch及cuda是那个版本,及显卡配置?

torch 版本:2.0.1+cu117,显卡配置:2块T4