Add evaluation with novel compositions

filippo-merlo · Feb 10, 2024 · a28c99a · a28c99a
1 parent 982f96d
commit a28c99a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/main.py b/main.py
@@ -288,6 +288,7 @@ def my_clip_train(in_path, out_path, model_name, source, in_base,
 	memory = pickle.load(infile)
 	infile.close()
 
+	# new stuff
 	# evaluate with novel compositions
 	evaluations['mare_novel_comp'].append(memory_evaluation(args.in_path, 'novel_test/', memory,
 						bn_n_test, ['rgba'], dic_train, vocabs))