e4f51e18508dc20ca379025108e59490dd59a6e0,eval_lm.py,,main,#Any#,17

Before Change

def main(args):
    assert args.path is not None, "--path required for evaluation!"

    args.tokens_per_sample = getattr(args, "tokens_per_sample", 1024)
    print(args)

    use_cuda = torch.cuda.is_available() and not args.cpu

After Change

def main(parsed_args):
    assert parsed_args.path is not None, "--path required for evaluation!"

    print(parsed_args)

    use_cuda = torch.cuda.is_available() and not parsed_args.cpu

    task = tasks.setup_task(parsed_args)

Instances