blblnbl

filippo-merlo · Feb 6, 2024 · 88418e7 · 88418e7
1 parent 27eb5a0
commit 88418e7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -289,11 +289,11 @@ def my_clip_train(in_path, out_path, model_name, source, in_base,
 	infile.close()
 
 	# evaluate with novel compositions
-	evaluations['mare_novel_comp'].append(memory_evaluation(args.in_path, 'novel_test/', memory,
+	evaluations['mare_novel_comp'].append(memory_evaluation(args.in_path, 'test/', memory,
 						bn_n_test, ['rgba'], dic_train, vocabs))
 
 	# evaluate with variations (dic_test)
-	evaluations['mare_var'].append(memory_evaluation(args.in_path, 'novel_test/', memory,
+	evaluations['mare_var'].append(memory_evaluation(args.in_path, 'test/', memory,
 						bn_n_train, ['rgba'], dic_test, vocabs))
 	pprint(evaluations)
 	with open(os.path.join(args.out_path, evaluations), 'w') as json_file: