Tensorboard恢复训练图
Tensorboard resume training plot
我 运行 一个强化学习训练脚本，它使用 Pytorch 并将数据记录到 tensorboardX 并保存检查点。现在我想继续训练。我如何告诉 tensorboardX 从我离开的地方继续？谢谢！
我想出了如何继续训练情节。在创建摘要编写器时，我们需要提供我们在第一次训练时使用的相同 log_dir。
from tensorboardX import SummaryWriter
writer = SummaryWriter('log_dir')
然后训练循环内的步骤需要从它离开的地方开始（而不是从0开始）：
writer.add_scalar('average reward',rewards.mean(),step)
我 运行 一个强化学习训练脚本，它使用 Pytorch 并将数据记录到 tensorboardX 并保存检查点。现在我想继续训练。我如何告诉 tensorboardX 从我离开的地方继续？谢谢！
我想出了如何继续训练情节。在创建摘要编写器时，我们需要提供我们在第一次训练时使用的相同 log_dir。
from tensorboardX import SummaryWriter
writer = SummaryWriter('log_dir')
然后训练循环内的步骤需要从它离开的地方开始（而不是从0开始）：
writer.add_scalar('average reward',rewards.mean(),step)