Number of parameter: 0.86M
Namespace(arch='hsir.model.qrnn3d.qrnn3d', bandwise=False, gpu_ids=[0], lr=None, name='hsir.model.qrnn3d.qrnn3d', noise='gaussian', resume=None, save_freq=10, save_root='checkpoints', schedule='hsir.schedule.denoise_default', test_root='data', train_root='data/ICVL64_31_100.db', use_conv2d=False)
Epoch [0] Use lr=0.001
Iters Elapsed Time Speed
0/3313 - -
0.0% | |Traceback (most recent call last):
File "hsirun/train.py", line 90, in
main()
File "hsirun/train.py", line 72, in main
if trainer.epoch < 30: trainer.train(train_loader1)
File "/home/tmpusr/miniconda3/envs/HSIR-1/lib/python3.8/site-packages/hsir/trainer.py", line 175, in train
for batch_idx, data in enumerate(train_loader):
File "/home/tmpusr/miniconda3/envs/HSIR-1/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 630, in next
data = self._next_data()
File "/home/tmpusr/miniconda3/envs/HSIR-1/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1344, in _next_data
return self._process_data(data)
File "/home/tmpusr/miniconda3/envs/HSIR-1/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1370, in _process_data
data.reraise()
File "/home/tmpusr/miniconda3/envs/HSIR-1/lib/python3.8/site-packages/torch/_utils.py", line 706, in reraise
raise exception
_pickle.UnpicklingError: Caught UnpicklingError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/home/tmpusr/miniconda3/envs/HSIR-1/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 309, in _worker_loop
data = fetcher.fetch(index) # type: ignore[possibly-undefined]
File "/home/tmpusr/miniconda3/envs/HSIR-1/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/tmpusr/miniconda3/envs/HSIR-1/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 52, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/tmpusr/miniconda3/envs/HSIR-1/lib/python3.8/site-packages/hsir/data/dataset.py", line 73, in getitem
img = self.dataset[index]
File "/home/tmpusr/miniconda3/envs/HSIR-1/lib/python3.8/site-packages/hsir/data/dataset.py", line 160, in getitem
x = pickle.loads(raw_datum)
_pickle.UnpicklingError: invalid load key, '\x08'.
一直显示这个问题，在文件夹中没有找到ICVL数据集的预处理文件
bit-isp
