2019年11月20日
pytorch 加载训练好的模型做inference
摘要： 前提： 模型参数和结构是分别保存的 1、 构建模型(# load model graph) model = MODEL() 2、加载模型参数(# load model state_dict) model.load_state_dict ( { k.replace('module.',''):v fo 阅读全文
posted @ 2019-11-20 10:52 U_C 阅读(3139) 评论(0) 推荐(0) 编辑