skip when xy =(0,0)

chris-chris · Jan 4, 2018 · d1ccbf2 · d1ccbf2
1 parent daded78
commit d1ccbf2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/common/vec_env/subproc_vec_env.py b/common/vec_env/subproc_vec_env.py
@@ -44,7 +44,7 @@ def worker(remote, map_name, i):
         #print("agent(",i," ) action : ", action2, " func : ", func, "xy :", action2[1][1])
         x, y = action2[1][1]
         move = True
-        if (x == -1 and y == -1):
+        if (x == 0 and y == 0):
           move = False
         result = env.step(actions=[action1])
         reward += result[0].reward

diff --git a/main.py b/main.py
@@ -43,8 +43,8 @@
 flags.DEFINE_boolean("prioritized", True, "prioritized_replay")
 flags.DEFINE_boolean("dueling", True, "dueling")
 flags.DEFINE_float("lr", 0, "Learning rate")
-flags.DEFINE_integer("num_agents", 2, "number of RL agents for A2C")
-flags.DEFINE_integer("num_scripts", 2, "number of script agents for A2C")
+flags.DEFINE_integer("num_agents", 20, "number of RL agents for A2C")
+flags.DEFINE_integer("num_scripts", 20, "number of script agents for A2C")
 flags.DEFINE_integer("nsteps", 20, "number of batch steps for A2C")
 
 PROJ_DIR = "/home/nsml/" #os.path.dirname(os.path.abspath(__file__))