Skip to content

Commit

Permalink
pprint
Browse files Browse the repository at this point in the history
  • Loading branch information
filippo-merlo committed Feb 6, 2024
1 parent 446202c commit 95c6e96
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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)

Expand Down

0 comments on commit 95c6e96

Please sign in to comment.