Skip to content

GitLab

"docs/git@developer.sourcefind.cn:OpenDAS/pytorch3d.git" did not exist on "09992a388ff5e14dde8110e61dbe4ba69cb3967e"
Commit 0e9e7f7b authored by alexeib's avatar alexeib Committed by Myle Ott
Browse files

make model access saner

parent d494485f
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
eval_lm.py
View file @ 0e9e7f7b
...@@ -37,11 +37,13 @@ def main(args): ...@@ -37,11 +37,13 @@ def main(args):
if args.fp16: if args.fp16:
model.half() model.half()
assert len(models) > 0
itr = data.EpochBatchIterator( itr = data.EpochBatchIterator(
dataset=task.dataset(args.gen_subset), dataset=task.dataset(args.gen_subset),
max_tokens=args.max_tokens, max_tokens=args.max_tokens,
max_sentences=args.max_sentences or 4, max_sentences=args.max_sentences or 4,
max_positions=model.max_positions(), max_positions=models[0].max_positions(),
num_shards=args.num_shards, num_shards=args.num_shards,
shard_id=args.shard_id, shard_id=args.shard_id,
ignore_invalid_inputs=True, ignore_invalid_inputs=True,
......
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