Ank*_*eka 8 tensorboard pytorch tensorboardx
我运行了一个强化学习训练脚本，该脚本使用 Pytorch 并将数据记录到 tensorboardX 并保存检查点。现在我想继续训练。我如何告诉tensorboardX从我离开的地方继续？谢谢你！
我想出了如何继续训练情节。在创建摘要编写器时，我们需要提供与log_dir第一次训练时使用的相同的内容。
from tensorboardX import SummaryWriter
writer = SummaryWriter('log_dir')
然后在训练循环中，步骤需要从离开的地方开始（而不是从 0 开始）：
writer.add_scalar('average reward',rewards.mean(),step)
| 归档时间： |
|
| 查看次数： |
3109 次 |
| 最近记录： |