深度學習 SODIC新賽 NLP技術實戰(zhàn)-企業(yè)隱患排查（文本挖掘）
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()
標簽：
