search

JXingZhao / EGNet

EGNet:Edge Guidance Network for Salient Object Detection (ICCV 2019)
311 stars 62 forks source link

batch_size can only be 1? #10

Open wangmunan opened 4 years ago

wangmunan commented 4 years ago

When I set batch_size = 8, I get problem: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.

JXingZhao commented 4 years ago

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.