From 5de2e2fb76d90788e21271ee63692cea0c3085a1 Mon Sep 17 00:00:00 2001 From: Danny Friar Date: Thu, 13 Jul 2023 21:29:54 +0100 Subject: [PATCH] Call `lazy_reset_grad` in on-device training docs (#16696) --- docs/python/on_device_training/training_api.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/python/on_device_training/training_api.rst b/docs/python/on_device_training/training_api.rst index cbc0d77e0a..64f81f3f18 100644 --- a/docs/python/on_device_training/training_api.rst +++ b/docs/python/on_device_training/training_api.rst @@ -32,7 +32,7 @@ Sample usage: module.train() training_loss = module(...) optimizer.step() - module.lazy_reset_grad + module.lazy_reset_grad() # Eval module.eval()