how to set batch size > 1 when trainDataset is ShangHaiA

leeyeehoo / CSRNet-pytorch

CSRNet: Dilated Convolutional Neural Networks for Understanding the Highly Congested Scenes

642 stars 259 forks source link

Open aboy2018 opened 4 years ago

aboy2018 commented 4 years ago

Because ShangHaiA dataset' size is not the same.How to set batch size >1 when train the code?

YuYue26 commented 3 years ago

you should change the cropsize to a fixed number