I am a newbie to Pytorch and I’m trying to build a cat dog classifier by training a model with the VGG16 architecture. However, when I run my model, it doesn’t really learn anything and always outputs either all ones or all zeros in any given batch in the test set which are a mix of cat and dog.
I have been stuck on this for a while, if anyone could give me any help. It would be much appreciated.
Code:
https://colab.research.google.com/drive/1WbixZzZYiRPASMVlqbV3br8_axVipHpG 3
Please find it here. Thanks a lot in advance.