make model access saner

0e9e7f7b · alexeib · Myle Ott · d494485f · 0e9e7f7b
Commit 0e9e7f7b authored Jul 02, 2018 by alexeib Committed by Myle Ott Jul 25, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

eval_lm.py eval_lm.py +3 -1

No files found.
--- a/eval_lm.py
+++ b/eval_lm.py
@@ -37,11 +37,13 @@ def main(args):
        if args.fp16:
            model.half()

+    assert len(models) > 0
+
    itr = data.EpochBatchIterator(
        dataset=task.dataset(args.gen_subset),
        max_tokens=args.max_tokens,
        max_sentences=args.max_sentences or 4,
-        max_positions=model.max_positions(),
+        max_positions=models[0].max_positions(),
        num_shards=args.num_shards,
        shard_id=args.shard_id,
        ignore_invalid_inputs=True,