Skip to content

Commit

Permalink
try
Browse files Browse the repository at this point in the history
  • Loading branch information
filippo-merlo committed Feb 6, 2024
1 parent db8ab5c commit 446202c
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
import torch.nn as nn
import torch.optim as optim
from PIL import Image
import json

from torch.utils.data import DataLoader

Expand Down Expand Up @@ -294,6 +295,9 @@ def my_clip_train(in_path, out_path, model_name, source, in_base,
evaluations['mare_var'].append(memory_evaluation(args.in_path, 'novel_test/', memory,
bn_n_train, ['rgba'], dic_test, vocabs))

with open(os.path.join(args.out_path, evaluations), 'w') as json_file:
json.dump(evaluations, json_file)

## Train for new word acquisition
## without
#model_name = 'nw_acquisition_old.pickle'
Expand Down

0 comments on commit 446202c

Please sign in to comment.