Keras to pytorch CNN is not working - dogs vs cats classification



Hi i have build simple CNN classifier to classifier dogs vs cats in keras keras notebook

I am very new to pytorch. I am trying to implement same keras network in pytorch. it doesn’t seem to be working. Can someone please help me point out what am i missing here.
pytorch notebook