fix total_len

JunhaoWang · Feb 19, 2019 · a2db811 · a2db811
1 parent b58c756
commit a2db811
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/babyai/imitation.py b/babyai/imitation.py
@@ -350,7 +350,7 @@ def initial_status():
             if status['i'] % self.args.log_interval == 0:
                 total_ellapsed_time = int(time.time() - total_start_time)
 
-                fps = total_len / (update_end_time - update_start_time)
+                fps = log['frames'] / (update_end_time - update_start_time)
                 duration = datetime.timedelta(seconds=total_ellapsed_time)
 
                 for key in log: