pytorch加载训练好的模型用来测试或者处理-finetuning-微调-转移学习

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

有空看

posted @ 2021-08-23 20:54  ice猫猫3  阅读(460)  评论(0编辑  收藏  举报