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))