python - stable_baselines 3 不存储 tensorboard_log

Question

我刚刚进入强化学习。我的模型不会在给定目录中创建任何文件。我究竟做错了什么？

def train():
    model = PPO('MlpPolicy', env, verbose=1, tensorboard_log=log_path)

    model.learn(total_timesteps=3000)
    with open(os.path.join(log_path,"Text.txt")) as file: #Test if given directory works
        print(file.read())

    model.save(PPO_Path)

经过训练的模型确实可以正常工作。

Answer 1

model.learn()函数应该有额外的参数tb_log_name

像这样：

model.learn(total_timesteps=3000, tb_log_name="experiment_1")