pytorch加载训练好的模型用来测试或者处理
训练函数部分的代码注释掉吗??,因为直接加载了.pt的参数文件
torch.load_state_dict()
PyTorch:存储和恢复模型并查看参数,load_state_dict(),state_dict()
pytorch加载训练好的模型用来测试或者处理
训练函数部分的代码注释掉吗??,因为直接加载了.pt的参数文件
torch.load_state_dict()
PyTorch:存储和恢复模型并查看参数,load_state_dict(),state_dict()