docs: polish training description

PKU-Alignment · Aug 27, 2023 · 507e370 · 507e370
2 parents 94b5df5 + b4847b1
commit 507e370
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/source/usage/train.rst b/docs/source/usage/train.rst
@@ -11,7 +11,7 @@ To run the algorithms with default configuration, you need to specify the enviro
 .. code-block:: bash
 
     cd safepo/single_agent
-    python ppo_lag.py   task SafetyPointGoal1-v0   experiment ppo_lag_exp
+    python ppo_lag.py   --task SafetyPointGoal1-v0   --experiment ppo_lag_exp
 
 Then you can check the results in the ``runs/ppo_lag_exp`` folder.
 
@@ -23,7 +23,7 @@ The multi-agent algorithms running is similar to the single-agent algorithms. Fo
 .. code-block:: bash
 
     cd safepo/multi_agent
-    python mappolag.py   task Safety2x4AntVelocity-v0   experiment mappo_lag_exp
+    python mappolag.py  --task Safety2x4AntVelocity-v0  --experiment mappo_lag_exp
 
 Then you can check the results in the ``runs/mappo_lag_exp`` folder.