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

eval_lm.py

+1 −2
Changes for eval_lm.py: 1 added line, 2 removed lines.
Original line number Diff line number Diff line
@@ -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