S_Crossroad 2021-08-11 21:42 采纳率: 0%
浏览 40

训练过的模型加载完和没训练过一样!!求各位回答!


torch.save({'state_dict':model.state_dict()},'./Word_embdding_state_dict.pth')
 
model_dict = torch.load('./Word_embdding_state_dict.pth')
model.load_state_dict(model_dict['state_dict'])
model.eval()

这是保存和加载的代码

我尝试过训练完的时候立马保存加载,准确率是很高的。但是等到重新运行一遍的时候加载,就完全不对了,准确率特别低。

我一度认为是nn.Embedding函数生成的词向量的值每次都是随机的才导致这个问题。

实在不知道如何解决,求各位回答一下!!

  • 写回答

1条回答 默认 最新

  • 影醉阏轩窗 2021-08-12 09:38
    关注

    参数打印出来对比

    评论

报告相同问题?

问题事件

  • 创建了问题 8月11日

悬赏问题

  • ¥15 有关类的报错,在模块里调用另一个模块的方法下的变量时出错
  • ¥15 delphi7 HMAC_SHA256方式加密
  • ¥15 关于#qt#的问题:我想实现qcustomplot完成坐标轴
  • ¥15 下列c语言代码为何输出了多余的空格
  • ¥15 kali linux用wget archive.kali.org/archive-key.asc指令下载签名无效(失败)
  • ¥15 openHarmony 利用c++程序在dayu210开发板上实现拉取RTSP视频流并且在屏幕上显示
  • ¥15 GD32H757的can通信配置
  • ¥20 nist随机数测试的问题
  • ¥20 帮我解决这个项目,thank you各位程序员
  • ¥15 哪位能用ea软件做一下,有偿