Function cb2

imperative/python/test/unit/xla/test_xla_training.py:317–318

(_, grad)

            return grad / 2

        def cb2(_, grad):
            return F.clip(grad, -2e-2, 2e-2)

        gm = autodiff.GradManager().attach(model.parameters(), callbacks=[cb1, cb2])
        optimizer = AdamW(model.parameters(), lr=0.01)