lhjthp / HNCT

18 stars 8 forks source link

尝试将数据集换成其他数据集时出错,始终无法解决 #11

Open Bobodesu opened 5 months ago

Bobodesu commented 5 months ago

仿照项目中的div2k.py文件写了一个realsr.py文件,然后运行的时候会出现这个问题 Traceback (most recent call last): File "E:\study\Graduation\Code\HNCT-main\src\main.py", line 42, in main() File "E:\study\Graduation\Code\HNCT-main\src\main.py", line 36, in main t.train() File "E:\study\Graduation\Code\HNCT-main\src\trainer.py", line 49, in train for batch, (lr, hr, _,) in enumerate(self.loader_train): File "E:\Anaconda\envs\torch\lib\site-packages\torch\utils\data\dataloader.py", line 628, in next data = self._next_data() File "E:\Anaconda\envs\torch\lib\site-packages\torch\utils\data\dataloader.py", line 1333, in _next_data return self._process_data(data) File "E:\Anaconda\envs\torch\lib\site-packages\torch\utils\data\dataloader.py", line 1359, in _process_data data.reraise() File "E:\Anaconda\envs\torch\lib\site-packages\torch_utils.py", line 543, in reraise raise exception RuntimeError: Caught RuntimeError in DataLoader worker process 2. Original Traceback (most recent call last): File "E:\Anaconda\envs\torch\lib\site-packages\torch\utils\data_utils\worker.py", line 302, in _worker_loop data = fetcher.fetch(index) File "E:\Anaconda\envs\torch\lib\site-packages\torch\utils\data_utils\fetch.py", line 61, in fetch return self.collate_fn(data) File "E:\Anaconda\envs\torch\lib\site-packages\torch\utils\data_utils\collate.py", line 265, in default_collate return collate(batch, collate_fn_map=default_collate_fn_map) File "E:\Anaconda\envs\torch\lib\site-packages\torch\utils\data_utils\collate.py", line 143, in collate return [collate(samples, collate_fn_map=collate_fn_map) for samples in transposed] # Backwards compatibility. File "E:\Anaconda\envs\torch\lib\site-packages\torch\utils\data_utils\collate.py", line 143, in return [collate(samples, collate_fn_map=collate_fn_map) for samples in transposed] # Backwards compatibility. File "E:\Anaconda\envs\torch\lib\site-packages\torch\utils\data_utils\collate.py", line 120, in collate return collate_fn_map[elem_type](batch, collate_fn_map=collate_fn_map) File "E:\Anaconda\envs\torch\lib\site-packages\torch\utils\data_utils\collate.py", line 162, in collate_tensorfn out = elem.new(storage).resize(len(batch), list(elem.size())) RuntimeError: Trying to resize storage that is not resizable 又是会出现其他的报错,比如有个EOFError之类的,不知道是网络中某个部分对其他数据集有影响还是怎么的,目前找到的问题是好像某些图片经过裁切后得不到规定的256256大小,不知道该如何解决,请愿意交流的朋友联系我的QQ:1814914563