def METHOD_NAME(test_case):
    def _lr_fn(parameters):
        of_sgd = flow.optim.SGD(parameters, lr=0.001)
        step_lr = flow.optim.lr_scheduler.StepLR(of_sgd, step_size=7, gamma=0.1)
        return of_sgd, step_lr
    _test_linear_graph_train_with_lr_sch(test_case, 21, flow.device("cuda"), _lr_fn)
    _test_linear_graph_train_with_lr_sch(test_case, 21, flow.device("cpu"), _lr_fn)