Pytorch Simple Classification Of Soils