Pytorch Simple Classification Charts