Support for batch size more than 1?

KumapowerLIU / Rethinking-Inpainting-MEDFE

Rethinking Image Inpainting via a Mutual Encoder Decoder with Feature Equalizations. ECCV 2020 Oral

Other

371 stars 50 forks source link

Support for batch size more than 1? #6

Open jackjacktang opened 4 years ago

jackjacktang commented 4 years ago

Thank you for releasing the code. I noticed that the batch size seems to be fixed to 1, I have tried a larger batch size and it seems not to work due to a size problem in BASE module in PConv, is it possible to support for batch size more than 1?

KumapowerLIU commented 4 years ago

Yes, it is can support batch size than 1. But I am not rewrite the code recently, i will update it as soon as possible.