FishYuLi/BalancedGroupSoftmax

batchsize or image_size problem

yangzhaojason opened this issue · 0 comments

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.