From 88418e74dc58bfda69905ce46ef5dde2b6982fc3 Mon Sep 17 00:00:00 2001 From: Filippo Date: Tue, 6 Feb 2024 18:16:25 +0100 Subject: [PATCH] blblnbl --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 9e9bd70..c8c7c42 100644 --- 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: