Skip to content

GitLab

"...en/git@developer.sourcefind.cn:chenpangpang/diffusers.git" did not exist on "a91a273d0b6c1b47be5005e4cf163adcb36d1176"
Unverified Commit a121e05f authored by Mishig's avatar Mishig Committed by GitHub
Browse files

Update custom_diffusion.mdx (#3165) 

Add missing newlines for rendering the links correctly
parent 3979aac9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
docs/source/en/training/custom_diffusion.mdx
View file @ a121e05f
...@@ -279,9 +279,11 @@ You can also perform inference from one of the complete checkpoint saved during ...@@ -279,9 +279,11 @@ You can also perform inference from one of the complete checkpoint saved during
TODO. TODO.
## Set grads to none ## Set grads to none
To save even more memory, pass the `--set_grads_to_none` argument to the script. This will set grads to None instead of zero. However, be aware that it changes certain behaviors, so if you start experiencing any problems, remove this argument. To save even more memory, pass the `--set_grads_to_none` argument to the script. This will set grads to None instead of zero. However, be aware that it changes certain behaviors, so if you start experiencing any problems, remove this argument.
More info: https://pytorch.org/docs/stable/generated/torch.optim.Optimizer.zero_grad.html More info: https://pytorch.org/docs/stable/generated/torch.optim.Optimizer.zero_grad.html
## Experimental results ## Experimental results
You can refer to [our webpage](https://www.cs.cmu.edu/~custom-diffusion/) that discusses our experiments in detail. You can refer to [our webpage](https://www.cs.cmu.edu/~custom-diffusion/) that discusses our experiments in detail.
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