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