search

lhjthp / HNCT

18 stars 8 forks source link

训练结果差距悬殊 #10

Open 1AnXu opened 8 months ago

1AnXu commented 8 months ago

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

1AnXu commented 8 months ago

我使用的命令如下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 ，作者能提供您们的训练命令吗

Liiiiaictx commented 4 months ago

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

1AnXu commented 4 months ago

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

3~4 days

Liiiiaictx commented 4 months ago

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

1AnXu commented 4 months ago

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

torch 版本：2.0.1+cu117，显卡配置：2块T4