Pytorch Print Model Weight