3

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

4

1 回答 1

6

我想出了如何继续训练情节。在创建摘要编写器时，我们需要提供与log_dir第一次训练时使用的相同的内容。

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

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

writer.add_scalar('average reward',rewards.mean(),step)
于 2020-09-11T13:11:43.687 回答