logo

agi-brain · Oct 23, 2023 · 7e161db · 7e161db
1 parent 09677a7
commit 7e161db
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/demo.py b/demo.py
@@ -5,8 +5,8 @@
 def parse_args():
     parser = argparse.ArgumentParser("Run a demo.")
     parser.add_argument("--method", type=str, default="a2c")
-    parser.add_argument("--env", type=str, default="atari")
-    parser.add_argument("--env-id", type=str, default="ALE/Pong-v5")
+    parser.add_argument("--env", type=str, default="classic_control")
+    parser.add_argument("--env-id", type=str, default="CartPole-v1")
     parser.add_argument("--test", type=int, default=0)
     parser.add_argument("--device", type=str, default="cuda:0")
     return parser.parse_args()

diff --git a/docs/source/figures/logo_1.png b/docs/source/figures/logo_1.png