From 95c6e96d9029be421c53e10b8d2aef5259b605a1 Mon Sep 17 00:00:00 2001 From: Filippo Date: Tue, 6 Feb 2024 17:14:39 +0100 Subject: [PATCH] pprint --- main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 3cb1e34..aad03db 100644 --- a/main.py +++ b/main.py @@ -9,6 +9,7 @@ import torch.optim as optim from PIL import Image import json +from pprint import pprint from torch.utils.data import DataLoader @@ -294,7 +295,7 @@ def my_clip_train(in_path, out_path, model_name, source, in_base, # evaluate with variations (dic_test) evaluations['mare_var'].append(memory_evaluation(args.in_path, 'novel_test/', memory, bn_n_train, ['rgba'], dic_test, vocabs)) - + pprint(evaluations) with open(os.path.join(args.out_path, evaluations), 'w') as json_file: json.dump(evaluations, json_file)