Update main_task_retrieval.py

FIX Bug microsoft#27
hemanthkumar17 · Mar 22, 2023 · ba3e7fa · ba3e7fa
1 parent 8db88f3
commit ba3e7fa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/main_task_retrieval.py b/main_task_retrieval.py
@@ -440,6 +440,7 @@ def eval_epoch(args, model, test_dataloader, device, n_gpu):
             sim_matrix = np.concatenate(tuple(sim_matrix), axis=0)
         else:
             sim_matrix = _run_on_single_gpu(model, batch_list, batch_list, batch_sequence_output_list, batch_visual_output_list)
+            sim_matrix = np.concatenate(tuple(sim_matrix), axis=0)
 
     metrics = compute_metrics(sim_matrix)
     logger.info('\t Length-T: {}, Length-V:{}'.format(len(sim_matrix), len(sim_matrix[0])))
@@ -512,4 +513,4 @@ def main():
             eval_epoch(args, model, test_dataloader, device, n_gpu)
 
 if __name__ == "__main__":
-    main()
+    main()