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