AttributeError: Caught AttributeError in DataLoader worker process 0.问题解决

多线程问题,把num_workers改成0即可解决。

希望对大家有帮助,欢迎评论区讨论!

[rank0]: Traceback (most recent call last): [rank0]: File "main_train.py", line 8, in <module> [rank0]: train_pipeline(root_path) [rank0]: File "/home/h00488860/anaconda3/envs/basicsr2/lib/python3.8/site-packages/ispbase/train.py", line 177, in train_pipeline [rank0]: prefetcher = CUDAPrefetcher(train_loader, opt) [rank0]: File "/home/h00488860/anaconda3/envs/basicsr2/lib/python3.8/site-packages/ispbase/data/prefetch_dataloader.py", line 100, in __init__ [rank0]: self.preload() [rank0]: File "/home/h00488860/anaconda3/envs/basicsr2/lib/python3.8/site-packages/ispbase/data/prefetch_dataloader.py", line 104, in preload [rank0]: self.batch = next(self.loader) # self.batch is a dict [rank0]: File "/home/h00488860/anaconda3/envs/basicsr2/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 630, in __next__ [rank0]: data = self._next_data() [rank0]: File "/home/h00488860/anaconda3/envs/basicsr2/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1344, in _next_data [rank0]: return self._process_data(data) [rank0]: File "/home/h00488860/anaconda3/envs/basicsr2/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1370, in _process_data [rank0]: data.reraise() [rank0]: File "/home/h00488860/anaconda3/envs/basicsr2/lib/python3.8/site-packages/torch/_utils.py", line 706, in reraise [rank0]: raise exception [rank0]: AttributeError: Caught AttributeError in DataLoader worker process 0. [rank0]: Original Traceback (most recent call last): [rank0]: File "/home/h00488860/anaconda3/envs/basicsr2/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 309, in _worker_loop [rank0]: data = fetcher.fetch(index) # type: ignore[possibly-undefined] [rank0]: File "/home/h00488860/anaconda3/envs/basicsr2/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch [rank0]: data = [self.dataset[idx] for idx in possibly_batched_index] [rank0]: File "/home/h00488860/anaconda3/envs/basicsr2/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 52, in <listcomp> [rank0]: data = [self.dataset[idx] for idx in possibly_batched_index] [rank0]: File "/home/h00488860/anaconda3/envs/basicsr2/lib/python3.8/site-packages/ispbase/data/comp_dataset.py", line 112, in __getitem__ [rank0]: item = self.dataset_list[dataset_id][data_id] [rank0]: File "/home/h00488860/chenxiong2/rdp_nsv/rdp_nsv_0513/isp_nsv/data/Nsv_Rdp_BigMonster_v0_dataset.py", line 723, in __getitem__ [rank0]: video_patch, gt, meta_info = self.unprocessing(clean_bgr) [rank0]: File "/home/h00488860/chenxiong2/rdp_nsv/rdp_nsv_0513/isp_nsv/data/dataset_common_utils.py", line 2180, in __call__ [rank0]: if self.dark_area_fluc : [rank0]: AttributeError: 'Unprocessing' object has no attribute 'dark_area_fluc'
最新发布
07-03
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值