问登机牌恢复训练区
EN

Stack Overflow用户

提问于 2020-09-10 22:24:09

回答 1查看 1.9K关注 0票数 4

我运行了一个强化学习培训脚本，它使用Pytorch并将数据记录到tensorboardX并保存了检查点。现在我想继续训练。我如何告诉tensorboardX从我停止的地方继续？谢谢!

回答已采纳

发布于 2020-09-11 13:11:43

我想出了如何继续训练的方法。在创建总结器时，我们需要提供与第一次培训时使用的相同的log_dir。

from tensorboardX import SummaryWriter
writer = SummaryWriter('log_dir')

然后，在训练循环步骤中，需要从它离开的地方开始(而不是从0开始)：

writer.add_scalar('average reward',rewards.mean(),step)

票数 7

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/63838417

复制

相似问题

问登机牌恢复训练区EN