fix number of update steps calculation in run script

inoryy · inoryy · commit 99df7c6d3e5c · 2018-11-25T18:29:35.000+02:00
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__':
