search

FishYuLi / BalancedGroupSoftmax

CVPR 2020 oral paper: Overcoming Classifier Imbalance for Long-tail Object Detection with Balanced Group Softmax.
Apache License 2.0
352 stars 64 forks source link

batchsize or image_size problem #24

Open yangzhaojason opened 2 years ago

yangzhaojason commented 2 years ago

my config: imgs_per_gpu=8, workflow: [('train', 1), ('val', 1)]

error: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.