如何在 Stable-Baseline、model.learn() 函数中设置 total_timesteps？答案

【问题标题】：How to set total_timesteps in Stable-Baseline, model.learn() function?如何在 Stable-Baseline、model.learn() 函数中设置 total_timesteps？
【发布时间】：2022-11-08 11:10:38
【问题描述】：

我正在使用 Stable-Baseline 来训练 A2C 模型。

我的数据长度是 9000。那么我应该在model.learn 中设置多少个total_timesteps？

model.learn(total_timesteps = 9000) # ?

我做了一些研究，有些建议像 10000，有些建议 100 万。我真的很困惑。

有什么建议么？

【问题讨论】：

不是一个编程问题，因此在这里题外话；请参阅stackoverflow.com/tags/reinforcement-learning/info 中的注释。另外，为什么要同时使用 pytorch 和 tensorflow 标签？

标签： tensorflow machine-learning pytorch reinforcement-learning stable-baselines

【解决方案1】：

Stable-Baseline 中的 total_timesteps 是要训练的样本总数（环境步数）。因此，它不应该与您拥有的剧集长度相同。

【讨论】：