Skip to content

GitLab

"torchvision/csrc/cpu/decoder/decoder.cpp" did not exist on "8b9859d3aeebcd37e6a284fc751c58569857f7be"
Commit 6eda8e47 authored by alexeib's avatar alexeib Committed by Myle Ott
Browse files

fix model loading in eval_lm

parent 295ccee9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
eval_lm.py
View file @ 6eda8e47
...@@ -26,8 +26,8 @@ def main(args): ...@@ -26,8 +26,8 @@ def main(args):
dataset = data_loaders.load_dataset(args, [args.gen_subset], False) dataset = data_loaders.load_dataset(args, [args.gen_subset], False)
# Load ensemble # Load ensemble
print('| loading model(s) from {}'.format(', '.join(args.path))) print('| loading model(s) from {}'.format(args.path))
models, _ = utils.load_ensemble_for_inference(args.path, dataset.src_dict, dataset.dst_dict) models, _ = utils.load_ensemble_for_inference(args.path.split(','), dataset.src_dict, dataset.dst_dict)
print('| Dictionary: {} types'.format(len(dataset.src_dict))) print('| Dictionary: {} types'.format(len(dataset.src_dict)))
print('| {} {} {} examples'.format(args.data, args.gen_subset, len(dataset.splits[args.gen_subset]))) print('| {} {} {} examples'.format(args.data, args.gen_subset, len(dataset.splits[args.gen_subset])))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment