2y7c3 / Super-Resolution-Neural-Operator

Super-Resolution Neural Operator, in CVPR 2023
MIT License
137 stars 12 forks source link

I encounter the following error reasons during testing. May I know why? #10

Open JeffreyYANGS opened 4 months ago

JeffreyYANGS commented 4 months ago

Traceback (most recent call last): File "F:\pythonProject1\Super-Resolution-Neural-Operator-main\test.py", line 122, in res = eval_psnr(loader, model, ^^^^^^^^^^^^^^^^^^^^^^^^ File "F:\pythonProject1\Super-Resolution-Neural-Operator-main\test.py", line 62, in eval_psnr for batch in pbar: File "F:\pythonProject1\venv\Lib\site-packages\tqdm\std.py", line 1182, in iter for obj in iterable: File "F:\pythonProject1\venv\Lib\site-packages\torch\utils\data\dataloader.py", line 630, in next data = self._next_data() ^^^^^^^^^^^^^^^^^ File "F:\pythonProject1\venv\Lib\site-packages\torch\utils\data\dataloader.py", line 1345, in _next_data return self._process_data(data) ^^^^^^^^^^^^^^^^^^^^^^^^ File "F:\pythonProject1\venv\Lib\site-packages\torch\utils\data\dataloader.py", line 1371, in _process_data data.reraise() File "F:\pythonProject1\venv\Lib\site-packages\torch_utils.py", line 694, in reraise raise exception RuntimeError: Caught RuntimeError in DataLoader worker process 0. Original Traceback (most recent call last): File "F:\pythonProject1\venv\Lib\site-packages\torch\utils\data_utils\worker.py", line 308, in _worker_loop data = fetcher.fetch(index) ^^^^^^^^^^^^^^^^^^^^ return collate_fn_map[elem_type](batch, collate_fn_map=collate_fn_map) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "F:\pythonProject1\venv\Lib\site-packages\torch\utils\data_utils\collate.py", line 162, in collate_tensor_fn
return torch.stack(batch, 0, out=out) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: stack expects each tensor to be equal size, but got [3, 161, 256] at entry 0 and [3, 192, 256] at entry 1