gpt4 book ai didi

pytorch - Tensorboard 简历训练图

转载 作者:行者123 更新时间:2023-12-04 13:12:20 24 4
gpt4 key购买 nike

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

最佳答案

我想出了如何继续训练情节。在创建 summarywriter 时,我们需要提供相同的 log_dir我们第一次训练时使用的。

from tensorboardX import SummaryWriter
writer = SummaryWriter('log_dir')
然后在训练循环步骤中需要从它离开的地方开始(而不是从 0 开始):
writer.add_scalar('average reward',rewards.mean(),step)

关于pytorch - Tensorboard 简历训练图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63838417/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com