Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "91da96431f836d8865b416f7d4a6e19c92ec0a15"
Commit e87b5799 authored by Jan Schlüter's avatar Jan Schlüter Committed by mcarilli
Browse files

Include loss scaling in README code example (#523)

parent 1904e48d
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
README.md
View file @ e87b5799
...@@ -67,6 +67,9 @@ model, optimizer = amp.initialize(model, optimizer, opt_level=opt_level) ...@@ -67,6 +67,9 @@ model, optimizer = amp.initialize(model, optimizer, opt_level=opt_level)
# Train your model # Train your model
... ...
with amp.scale_loss(loss, optimizer) as scaled_loss:
scaled_loss.backward()
...
# Save checkpoint # Save checkpoint
checkpoint = { checkpoint = {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment