Pytorch Simple Classification Of Living