需要一个灵活的方法来在 TensorBoard 中记录训练数据
作者: Xiong5Heng创建于 2023年6月29日更新于 2025年12月12日
标签enhancementgood first issue
- 我已经标记了所有适用的类别:
- 发现错误 (引起异常)
- RL 算法错误
- 文档请求 (例如"文档中缺少 X。")
- 新功能请求
import tianshou, gymnasium as gym, torch, numpy, sys
print(tianshou.__version__, gym.__version__, torch.__version__, numpy.__version__, sys.version, sys.platform)
0.5.0 0.26.3 1.12.1 1.21.5 3.7.13 (default, Oct 18 2022, 18:57:03)
[GCC 11.2.0] linux您好, 感谢您的杰出工作! 在使用 Tianshou 时,我遇到了一个与在 TensorBoard 中记录训练数据有关的问题。 具体来说,我需要记录 info 字典中的一个值,例如 info["key"],这类似于每个回合结束时的累计奖励。
内容来源: thu-ml/tianshou