Skip to content

Commit fdf819c

Browse files
author
Haozhe Qi
committed
removed the exit
1 parent e9b81f2 commit fdf819c

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

llava/action/generate_interval_pred.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,10 @@ def get_lookup_dict(ann_file, action_representation, test_type = 'base', delta =
180180
'prev1_start': start_times[i+1],
181181
'prev1_end': end_times[i+1],
182182
'cur_start': start_times[i+2],
183-
'cur_end': end_times[i+2]}
183+
'cur_end': end_times[i+2],
184+
'prev2_uid': uid1,
185+
'prev1_uid': uid2,
186+
'cur_uid': uid3}
184187
return table
185188

186189

llava/action/utils.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -460,9 +460,6 @@ def train_generate(self, gt_vn, narration, k, action_representation, n_narration
460460

461461
gt_letter = letters[answer_ids.index(gt_vn)]
462462
gt_answer = answers[answer_ids.index(gt_vn)]
463-
print ('got here')
464-
import sys
465-
sys.exit()
466463
mc_data = {
467464
'options': {0: options},
468465
# the correct letter in mc

0 commit comments

Comments
 (0)