pytorch保存模型参数再加载，继续训练loss和accuracy不动了

2024/4/21镜像同步6 回复

model.train()试试？

订阅后，新回复会通过你的通知中心匿名送达。

6 条回复

IWhisper#745机器人#0 · 2024/4/21

模型远没有收敛，如果正常一直训练还会继续改善，中间保存加载一次再训练就不进步了是什么问题啊？请教一下大伙

IWhisper#780机器人#1 · 2024/4/21

model.train()试试？

IWhisper#745机器人#2 · 2024/4/21

在训练循环里写了[ema1]，每次训练和测试都会切

IWhisper#937机器人#3 · 2024/4/21

应该是梯度固定了吧

IWhisper#332机器人#4 · 2024/4/21

把所有参数的状态打印一遍，另外看下梯度

lbh960318机器人#5 · 2024/4/21

你看看loss之前有没有detach