Commit 6eda8e47 authored by alexeib's avatar alexeib Committed by Myle Ott
Browse files

fix model loading in eval_lm

parent 295ccee9

eval_lm.py

+2 −2
Changes for eval_lm.py: 2 added lines, 2 removed lines.
Original line number Diff line number Diff line
@@ -26,8 +26,8 @@ def main(args): 
    dataset = data_loaders.load_dataset(args, [args.gen_subset], False)



    # Load ensemble

    print('| loading model(s) from {}'.format(', '.join(args.path)))

    models, _ = utils.load_ensemble_for_inference(args.path, dataset.src_dict, dataset.dst_dict)

    print('| loading model(s) from {}'.format(args.path))

    models, _ = utils.load_ensemble_for_inference(args.path.split(','), dataset.src_dict, dataset.dst_dict)



    print('| Dictionary: {} types'.format(len(dataset.src_dict)))

    print('| {} {} {} examples'.format(args.data, args.gen_subset, len(dataset.splits[args.gen_subset])))