saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-03 03:58:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Call lazy_reset_grad in on-device training docs (#16696)

Browse source
This commit is contained in:
Danny Friar 2023-07-13 21:29:54 +01:00 committed by GitHub
parent a461608409
commit 5de2e2fb76
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
docs/python/on_device_training/training_api.rst
View file

@ -32,7 +32,7 @@ Sample usage:
module.train()
training_loss = module(...)
optimizer.step()
module.lazy_reset_grad
module.lazy_reset_grad()
# Eval
module.eval()