Pytorch使用问题记录

主要记录使用pytorch过程中的一些问题及解决方法
1、GPU训练网络后，在CPU中使用的问题
1.1、使用load_static_dict API时需要根据是否使用GPU指定map参数。

if self.num_gpu == 0:
    map_location = lambda storage, loc: storage
else:
    map_location = None

1.2、在我的一次实验中，多GPU训练后保存的pth模型和单GPU的有点不太一致，使得多GPU的模型不能在CPU上加载。解决方案是，先在多GPU训练，最后几个epoch使用单GPU训练，之后保存单GPU的训练模型。