Sue-Tang-Up
commented
8 months ago I find the same problem. Have you ever solved that?
Open jiubie12 opened 1 year ago
Sue-Tang-Up
commented
8 months ago I find the same problem. Have you ever solved that?
This error occurred during training. How to solve it? Traceback (most recent call last): File "tools/train.py", line 128, in
main()
File "tools/train.py", line 124, in main
timestamp=timestamp)
File "/root/autodl-tmp/SOLO/mmdet/apis/train.py", line 111, in train_detector
timestamp=timestamp)
File "/root/autodl-tmp/SOLO/mmdet/apis/train.py", line 297, in _non_dist_train
runner.run(data_loaders, cfg.workflow, cfg.total_epochs)
File "/root/miniconda3/envs/solo/lib/python3.7/site-packages/mmcv/runner/epoch_based_runner.py", line 127, in run
epoch_runner(data_loaders[i], **kwargs)
File "/root/miniconda3/envs/solo/lib/python3.7/site-packages/mmcv/runner/epoch_based_runner.py", line 47, in train
for i, data_batch in enumerate(self.data_loader):
File "/root/miniconda3/envs/solo/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 517, in next
data = self._next_data()
File "/root/miniconda3/envs/solo/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1199, in _next_data
return self._process_data(data)
File "/root/miniconda3/envs/solo/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1225, in _process_data
data.reraise()
File "/root/miniconda3/envs/solo/lib/python3.7/site-packages/torch/_utils.py", line 429, in reraise
raise self.exc_type(msg)
TypeError: Caught TypeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/root/miniconda3/envs/solo/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 202, in _worker_loop
data = fetcher.fetch(index)
File "/root/miniconda3/envs/solo/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/miniconda3/envs/solo/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/autodl-tmp/SOLO/mmdet/datasets/custom.py", line 131, in getitem
data = self.prepare_train_img(idx)
File "/root/autodl-tmp/SOLO/mmdet/datasets/custom.py", line 144, in prepare_train_img
return self.pipeline(results)
File "/root/autodl-tmp/SOLO/mmdet/datasets/pipelines/compose.py", line 24, in call
data = t(data)
File "/root/autodl-tmp/SOLO/mmdet/datasets/pipelines/transforms.py", line 314, in call
self._pad_masks(results)
File "/root/autodl-tmp/SOLO/mmdet/datasets/pipelines/transforms.py", line 301, in _pad_masks
for mask in results[key]
File "/root/autodl-tmp/SOLO/mmdet/datasets/pipelines/transforms.py", line 301, in
for mask in results[key]
TypeError: impad() takes 1 positional argument but 2 positional arguments (and 1 keyword-only argument) were given