Function train

imperative/python/test/integration/test_trace_dump.py:172–178

(d)

    @trace(symbolic=True)
    def train(d):
        with gm:
            loss = model(d)
            gm.backward(loss)

        optim.step().clear_grad()
        return loss

    @trace(symbolic=True)
    def val(d):