sizes of tensors don't match when running the testing code
Hi Max,
When I tried to test the code on the LIVE1 dataset, the following error occurs. Testing on other datasets would have a similar error; Only a few images can generate a result. Could you help me with this problem? Thank you!
python experiments/correct_batch.py --input testsets/LIVE1_Color/10/ --output results/LIVE1_Color/10/ --weights ./weights/weights_c.pt --dstats_path ./stats/cstats.pt --color
Correcting: 10% 3/29 [00:03<00:29, 1.12s/batch] Traceback (most recent call last): File "experiments/correct_batch.py", line 60, in y_r, cb_r, cr_r = net(y_q, y, c_q, cb, cr) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 722, in _call_impl result = self.forward(*input, **kwargs) File "./model/deartifacting_network.py", line 88, in forward r_cb = self.color_net(q_y, q_c, restored, cb) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 722, in _call_impl result = self.forward(*input, **kwargs) File "./model/color_restore.py", line 42, in forward d = self.block_enhancer2(torch.cat([d, y], dim=1)) RuntimeError: Sizes of tensors must match except in dimension 2. Got 61 and 62 (The offending index is 0)