Pytorch(4)-模型保存-载入-eval()1. 整个模型 保存-载入2. 仅模型参数 保存-载入3. GPU/CPU模型保存与导入4. net.eval()–固定模型随机项
模型保存与提取1.网络保存2. 网络提取3.GPU/CPU模型保存与导入神经网络模型在线训练完之后需要保存下来,以便下次使用时可以直接导入已经训练好的模型。pytorch 提供两种方式保存模型,一种是直接保存整个模型,另一种只保存模型参数。1.网络保存torch.save(net1, ‘net.pkl’) # 保存整个网络torch.save(net1.state_dict(), ...