From f0a2d821c827134671c944c87b45a61cafbc4b52 Mon Sep 17 00:00:00 2001 From: Filippo Date: Thu, 8 Feb 2024 17:53:38 +0100 Subject: [PATCH] bo --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index f773c6b..41fd14b 100644 --- a/main.py +++ b/main.py @@ -289,7 +289,7 @@ 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, 'test/', memory, + evaluations['mare_novel_comp'].append(memory_evaluation(args.in_path, 'novel_test/', memory, bn_n_test, ['rgba'], dic_train, vocabs)) # evaluate with variations (dic_test)