Skip to content

Commit 4e19fa9

Browse files
author
Elad Hoffer
committed
fix eval
1 parent 110266b commit 4e19fa9

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

eval.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -76,10 +76,12 @@ def main(args):
7676

7777
checkpoint = torch.load(args.checkpoint , map_location=lambda storage, loc: storage)
7878
config = checkpoint['config']
79+
src_tok, target_tok = checkpoint['tokenizers'].values()
80+
7981
args.data_config = literal_eval(args.data_config)
8082
dataset = getattr(datasets, args.dataset)
83+
args.data_config['tokenizers'] = checkpoint['tokenizers']
8184
val_data = dataset(args.dataset_dir, split='dev', **args.data_config)
82-
src_tok, target_tok = checkpoint['tokenizers'].values()
8385

8486
model = getattr(models, config.model)(**config.model_config)
8587
model.load_state_dict(checkpoint['state_dict'])

0 commit comments

Comments
 (0)