unable to handle batchnorm1d in pytorch

waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.

MIT License

1.79k stars 266 forks source link

Open duhd1993 opened 5 years ago

duhd1993 commented 5 years ago

A simple example, no issues 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?