主要介绍了python PyTorch预训练示例,小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧

前言 最近使用PyTorch感觉妙不可言，有种当初使用Keras的快感，而且速度还不慢。各种设计直接简洁，方便研究，比tensorflow的臃肿好多了。今天让我们来谈谈PyTorch的预训练，主要是自己写代码的经验以及论坛PyTorch Forums上的一些回答的总结整理。 直接加载预训练模型 如果我们使用的模型和原模型完全一样，那么我们可以直接加载别人训练好的模型： my_resnet = MyResNet(*args, **kwargs) my_resnet.load_state_dict