Update eval_subgoals.py

gistvision · Dec 5, 2020 · 1a00910 · 1a00910
1 parent 05ec646
commit 1a00910
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/models/eval/eval_subgoals.py b/models/eval/eval_subgoals.py
@@ -221,6 +221,9 @@ def evaluate(cls, env, model, eval_idx, r_idx, resnet, traj_data, args, lock, su
             # increment time index
             t += 1
 
+            prev_image = curr_image
+            prev_action = action
+
         # metrics
         pl = float(t - len(expert_init_actions)) + 1 # +1 for last action
         expert_pl = len([ll for ll in traj_data['plan']['low_actions'] if ll['high_idx'] == eval_idx])