Pytorch Simple Classification Of Animals