bug fix

flow-project · Jun 9, 2020 · 4be98d3 · 4be98d3
1 parent ac7f955
commit 4be98d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/train.py b/examples/train.py
@@ -306,7 +306,7 @@ def train_h_baselines(flow_params, args, multiagent):
 
         # Perform training.
         alg.learn(
-            total_timesteps=args.total_steps,
+            total_steps=args.total_steps,
             log_dir=dir_name,
             log_interval=args.log_interval,
             eval_interval=args.eval_interval,