sy-eng
sy-eng
I tried the tutorial on Colab and now it looks the modified code works and the original code have the deprecation warning.
Thank you for your reply! I also retrained it and found very little difference between the models with and without pooling layer. 
Thank you for your comment. > I had tried a couple of trainings in PyTorch with 50 epochs and noticed that some start learning suddenly, but many continued to fail...