Pytorch Print Model