happywu / Sequence-Level-Semantics-Aggregation

Sequence Level Semantics Aggregation for Video Object Detection
Apache License 2.0
85 stars 19 forks source link

How can I use batch size > 1? #7

Closed TyroneLi closed 4 years ago

TyroneLi commented 4 years ago

When I try to use batch size greater than 1 for training, error occurs. If I want to change batch size bigger than 1, could you offer some advice how to modify code?Thx~

happywu commented 4 years ago

You need to modify the dataloader to provide data that has a batch size greater than 1. You need to pad images to make them have the same size in order to form mini batch. And also the symbol file since there is some operation that assume the batch size is 1. You should do shape verify for the symbol file.