Training is started.
Traceback (most recent call last):
File "mh_SR_net_1029/1202_9_layer_train_M_GPU.py", line 111, in <module>
main()
File "mh_SR_net_1029/1202_9_layer_train_M_GPU.py", line 104, in main
model.train(<
Sizes of tensors must match except in dimension 0. Got 339 and 387 问题解决
最新推荐文章于 2024-08-28 22:39:13 发布
博客解决了在处理4维输入时遇到的Tensor尺寸匹配问题。通过在`dataset.py`中添加打印语句定位问题,发现是由于在旧版本修改后未正确更新crop后的`img_LR`和`img_HR`值导致的。修复此错误后,问题得到解决。
摘要由CSDN通过智能技术生成