Cnn Model Python Pytorch Compatibility