FishAndWasabi / YOLO-MS

YOLO-MS: Rethinking Multi-Scale Representation Learning for Real-Time Object Detection
Other
222 stars 23 forks source link

TypeError: yolov5_collate() got an unexpected keyword argument '_scope_' #8

Closed CcccccaptionOvO closed 1 year ago

CcccccaptionOvO commented 1 year ago

Traceback (most recent call last): File "tools/train.py", line 114, in main() File "tools/train.py", line 110, in main runner.train() File "/data/.conda/envs/YOLO-MS/lib/python3.8/site-packages/mmengine/runner/runner.py", line 1706, in train model = self.train_loop.run() # type: ignore File "/data/.conda/envs/YOLO-MS/lib/python3.8/site-packages/mmengine/runner/loops.py", line 96, in run self.run_epoch() File "/data/.conda/envs/YOLO-MS/lib/python3.8/site-packages/mmengine/runner/loops.py", line 111, in run_epoch for idx, data_batch in enumerate(self.dataloader): File "/data/.conda/envs/YOLO-MS/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 681, in next data = self._next_data() File "/data/.conda/envs/YOLO-MS/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1376, in _next_data return self._process_data(data) File "/data/.conda/envs/YOLO-MS/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1402, in _process_data data.reraise() File "/data/.conda/envs/YOLO-MS/lib/python3.8/site-packages/torch/_utils.py", line 461, in reraise raise exception TypeError: Caught TypeError in DataLoader worker process 0. Original Traceback (most recent call last): File "/data/.conda/envs/YOLO-MS/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 302, in _worker_loop data = fetcher.fetch(index) File "/data/.conda/envs/YOLO-MS/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch return self.collate_fn(data) TypeError: yolov5_collate() got an unexpected keyword argument 'scope'

CcccccaptionOvO commented 1 year ago

我在'mmyolo::rtmdet/rtmdet_l_syncbn_fast_8xb32-300e_coco.py'文件中按照以下步骤修改了dataloader,然后出现了一些新的问题 image

image

FishAndWasabi commented 1 year ago

您好,不需要在mmyolo::rtmdet/rtmdet_l_syncbn_fast_8xb32-300e_coco.py进行修改。请在您所使用的config文件里进行修改。

感谢您对我们工作的关注!

Best Wishes! 😊

CcccccaptionOvO commented 1 year ago

非常感谢!!!