huawei-noah / Pretrained-IPT

Apache License 2.0
434 stars 63 forks source link

测试sr任务问题 #37

Closed oneonejia closed 2 years ago

oneonejia commented 2 years ago

在测试sr x2任务时我将scale设置为2,PSNR结果是正确的 但我测试sr x3时,将scale 设置为3的时候出现了 RuntimeError: Error(s) in loading state_dict for ipt: size mismatch for tail.0.0.0.weight: copying a param with shape torch.Size([256, 64, 3, 3]) from checkpoint, the shape in current model is torch.Size([576, 64, 3, 3]). size mismatch for tail.0.0.0.bias: copying a param with shape torch.Size([256]) from checkpoint, the shape in current model is torch.Size([576]). 测试sr x4 时,将scale 设置为4,但是PSNR仅为12.448

这是为什么呢