batch_size can only be 1?
Opened this issue · 1 comments
wangmunan commented
When I set batch_size = 8, I get problem:
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.
JXingZhao commented
Yeah, Because the sizes of different images are different.
So if you want to use big batch, you should resize the images.
发件人：wangmunan <notifications@github.com>
发送日期：2019-09-24 21:26:18
收件人：JXingZhao/EGNet <EGNet@noreply.github.com>
抄送人：Subscribed <subscribed@noreply.github.com>
主题：[JXingZhao/EGNet] batch_size can only be 1? (#10)
When I set batch_size = 8, I get problem:
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.
—
You are receiving this because you are subscribed to this thread.
Reply to this email directly, view it on GitHub, or mute the thread.