Pytorch：pytorch中的多种训练模型写法