search

DeniJsonC / WaveNet

[PG 2023] WaveNet: Wave-Aware Image Enhancement
Other
78 stars 14 forks source link

测试问题 #1

Closed cjl506 closed 9 months ago

cjl506 commented 10 months ago

image 测试的时候，提示这样的错误，该如何解决呢

DeniJsonC commented 10 months ago

image 测试的时候，提示这样的错误，该如何解决呢

你好，我在测试时没有发现该问题，你可以尝试在主函数里用以下两句，而不调用‘load_checkpoint’函数运行一下： checkpoint = torch.load($weights_path$, map_location=device) model.load_state_dict(checkpoint)