hiddenlayer icon indicating copy to clipboard operation
hiddenlayer copied to clipboard

unable to handle batchnorm1d in pytorch

Open duhd1993 opened this issue 4 years ago • 0 comments

A simple example, no issues image When added batchnorm1d between layers,

Expected more than 1 value per channel when training, got input size torch.Size([1, 100])

It seems like it's treating batchnorm1d as batchnorm2d?

duhd1993 avatar Jul 27 '19 19:07 duhd1993