Pytorch Simple Classification Of Bacteria