Removed training scores computation.

noagarcia · Dec 2, 2019 · 15a9f3b · 15a9f3b
1 parent ef703fe
commit 15a9f3b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/KnowledgeRetrieval/process.py b/KnowledgeRetrieval/process.py
@@ -206,5 +206,6 @@ def evaluate(args, outdir, split):
         evaluate(args, outdir, split='test')
     if not os.path.exists(os.path.join(args.data_dir, 'retieval_scores_val.pckl')):
         evaluate(args, outdir, split='val')
-    if not os.path.exists(os.path.join(args.data_dir, 'retieval_scores_train.pckl')):
-        evaluate(args, outdir, split='train')
+    # Do not compute scores for training, we use random samples in the VideoReasoning training part
+    # if not os.path.exists(os.path.join(args.data_dir, 'retieval_scores_train.pckl')):
+    #     evaluate(args, outdir, split='train')