From a28c99aa4133d0727dfdb30825cedf3e658838f9 Mon Sep 17 00:00:00 2001 From: Filippo Date: Sat, 10 Feb 2024 16:01:57 +0100 Subject: [PATCH] Add evaluation with novel compositions --- main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/main.py b/main.py index d240688..9bd0e97 100644 --- 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))