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")