Jongchan/attention-module

Can not run in batch size 1

Closed this issue · 1 comments

leezonpen commented

I try this module in detection task. When i try to make a inference, it throw an error. I found that it can not run forward in one batch size because of the nn.BatchNorm1d . How to deal with it?

leezonpen commented

I get it .
model.eval() should be used.