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()