hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

move doc string

Browse Source
This commit is contained in:
Maximilian Huettenrauch 2024-03-26 14:26:27 +01:00
parent d9a201754c
commit 2e3f0b5e99
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
examples/mujoco/mujoco_ppo_hl_multi.py
View File

@ -1,7 +1,4 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
"""Use the high-level API of TianShou to evaluate the PPO algorithm on a MuJoCo environment with multiple seeds for a
given configuration. After the agents are trained, the results are evaluated using rliable API.
"""
import os import os
from collections.abc import Sequence from collections.abc import Sequence
@ -53,6 +50,10 @@ def main(
norm_adv: bool = False, norm_adv: bool = False,
recompute_adv: bool = True, recompute_adv: bool = True,
) -> str: ) -> str:
"""Use the high-level API of TianShou to evaluate the PPO algorithm on a MuJoCo environment with multiple seeds for
a given configuration. The results for each run are stored in separate sub-folders. After the agents are trained,
the results are evaluated using rliable API.
"""
log_name = os.path.join("log", task, "ppo", datetime_tag()) log_name = os.path.join("log", task, "ppo", datetime_tag())
experiment_config.persistence_base_dir = log_name experiment_config.persistence_base_dir = log_name