python - 更改 Ray RLlib Training 的 Logdir 而不是 ~/ray_results

Question

我正在使用 Ray & RLlib 在 Ubuntu 系统上训练 RL 代理。 Tensorboard 用于通过指向 ~/ray_results 来监控训练进度所有运行的所有日志文件都存储在其中。未使用 Ray Tune。

例如，在开始新的 Ray/RLlib 训练运行时，将在以下位置创建一个新目录

~/ray_results/DQN_ray_custom_env_2020-06-07_05-26-32djwxfdu1

为了可视化训练进度，我们需要使用以下命令启动 Tensorboard

tensorboard --logdir=~/ray_results

问题:是否可以配置 Ray/RLlib 来更改日志文件的输出目录 ~/ray_results到另一个位置？

此外，不要登录到名为 DQN_ray_custom_env_2020-06-07_05-26-32djwxfdu1 之类的目录。 ，这个目录名可以自己设置吗？

尝试失败:尝试设置

os.environ['TUNE_RESULT_DIR'] = '~/another_dir`

运行前 ray.init() ，但结果日志文件仍在写入 ~/ray_results .

Answer 1

Is it possible to configure Ray/RLlib to change the output directory of the log files from ~/ray_results to another location?

目前无法使用 RLib CLI 工具 ( rllib ) 进行配置。
如果您对 Python API 没问题，那么，如 documentation 中所述, local_dir tune.run 的参数负责指定输出目录，默认为 ~/ray_results .

Additionally, instead of logging to a directory named something like DQN_ray_custom_env_2020-06-07_05-26-32djwxfdu1, can this directory name by set by ourselves?

这由 trial_name_creator 管理 tune.run 的参数.它必须是一个接受试验对象并将其格式化为字符串的函数，如下所示:

def trial_name_id(trial):
    return f"{trial.trainable_name}_{trial.trial_id}"

tune.run(...trial_name_creator=trial_name_id)