深度学习 SODIC新赛 NLP技术实战-企业隐患排查（文本挖掘）
2023-01-15 14:19 作者:janet19961217 | 我要投稿
如果只有保存的整个模型，也可以采用提取权重的方式构建新的模型：
# 读取整个模型
loaded_whole_model = torch.load(save_dir)
loaded_model = models.resnet152() #注意这里需要对模型结构有定义
loaded_model.state_dict = loaded_whole_model.state_dict
loaded_model = nn.DataParallel(loaded_model).cuda()
