about batch size

[superli6]

Hello author, when I set the batch size to 2, I encounter an error: "RuntimeError: stack expects each tensor to be equal size, but got [5120, 1024] at entry 0 and [13952, 1024] at entry 1." What is the reason for this? I'm looking forward to your reply.

[liupei101]

batch_size must be set to 1 as the bag often has a variable size and multiple size-variable inputs cannot be directly packed into a mini-batch for training.