fix number of update steps calculation in run script

inoryy · Nov 25, 2018 · 99df7c6 · 99df7c6
1 parent 5b22267
commit 99df7c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/reaver/run.py b/reaver/run.py
@@ -105,7 +105,7 @@ def main(argv):
         expt.save_gin_config()
         expt.save_model_summary(agent.model)
 
-    agent.run(env, args.updates)
+    agent.run(env, args.updates * args.traj_len * args.batch_sz // args.envs)
 
 
 if __name__ == '__main__':