2a15843223ac09c1331b1ca444da179ccab735d9,test/test_regression_1.py,,run,#,14

Before Change

    with torch.no_grad():
        outputs = model(inputs)
        torch.save(outputs, args.out_outputs)
        torch.save(model.state_dict(), args.out_model)


if __name__ == "__main__":

After Change

    with torch.no_grad():
        outputs = model(inputs)
    torch.save(inputs, args.out_inputs)
    torch.save(outputs, args.out_outputs)
    torch.save(model.state_dict(), args.out_model)


if __name__ == "__main__":

Instances