From b4847b11ce9a6f0ee2abbab94767016f931841bb Mon Sep 17 00:00:00 2001
From: Gaiejj <524339208@qq.com>
Date: Sun, 27 Aug 2023 23:51:25 +0800
Subject: [PATCH] docs: polish training description

---
 docs/source/usage/train.rst | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/source/usage/train.rst b/docs/source/usage/train.rst
index 79d11d6..1b4ca19 100644
--- 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.