Unverified Commit 3feaccf3 authored by Alexei Baevski's avatar Alexei Baevski Committed by GitHub
Browse files

Merge pull request #193 from hmc-cs-mdrissi/minor_fixes

default samples_per_token in eval_lm
parents b458977a 891f2307
......@@ -17,8 +17,7 @@ from fairseq.sequence_scorer import SequenceScorer
def main(args):
assert args.path is not None, '--path required for evaluation!'
if args.tokens_per_sample is None:
args.tokens_per_sample = 1024
args.tokens_per_sample = getattr(args, 'tokens_per_sample', 1024)
print(args)
use_cuda = torch.cuda.is_available() and not args.cpu
......
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