Cannot set batch_size>1?

megvii-research / MOTRv2

[CVPR2023] MOTRv2: Bootstrapping End-to-End Multi-Object Tracking by Pretrained Object Detectors

Other

343 stars 44 forks source link

Cannot set batch_size>1? #24

Closed wh1sperC closed 1 year ago

wh1sperC commented 1 year ago

When I set batch_size=2 to train my model, it told me "AttributeError: 'list' object has no attribute 'requires_grad'."So how to solve it to enlarge batch_size?I want to save my time on training model.

zyayoung commented 1 year ago

We do not support batch_size>1 at the moment. We will follow up in #13 for updates.