I tried to implement a convolutional auto encoder in pytorch and applied it on MNIST. Unfortunately the train loss makes almost no progress (it does not decrease) during training. Can someone please help? I have posted my code (< 95 lines) at the following Github Gist: code is here 26.