search

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_' #10

Closed Bonasdljlkj closed 11 months ago

Bonasdljlkj commented 1 year ago

你好，我在训练过程中遇到该报错，请问有解决办法嘛，谢谢。TypeError: yolov5_collate() got an unexpected keyword argument 'scope'

Post-Station-Life commented 1 year ago

你的这个问题解决了吗？，如何解决的

FishAndWasabi commented 1 year ago

不好意思，没有及时回复您。该报错需要在config里的dataloader里做一些处理（我们已在commit 5499293 进行修复）:

train_dataloader = dict(
    ...
    collate_fn=dict(_delete_=True, type='yolov5_collate'),
    sampler=dict(_delete_=True, type='DefaultSampler', shuffle=True),
    ...
)

val_dataloader = dict(
    ...
    sampler=dict(_delete_=True, type='DefaultSampler', shuffle=True),
    ...
)

test_dataloader = val_dataloader

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

Best Wishes! 😊