diff --git a/main.py b/main.py index f4ecbc7..c48f10b 100644 --- a/main.py +++ b/main.py @@ -296,7 +296,7 @@ def my_clip_train(in_path, out_path, model_name, source, in_base, evaluations['mare_var'].append(memory_evaluation(args.in_path, 'novel_train/', memory, bn_n_train, ['rgba'], dic_test, vocabs)) pprint(evaluations) - with open(os.path.join(args.out_path, evaluations), 'w') as json_file: + with open(os.path.join(args.out_path, 'evaluations.json'), 'w') as json_file: json.dump(evaluations, json_file) ## Train for new word acquisition