AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Update custom_diffusion.mdx (#3165)

Browse Source 
Add missing newlines for rendering the links correctly
This commit is contained in:
Mishig
2023-04-20 11:04:06 +02:00
committed by GitHub
parent 3979aac996
commit a121e05feb
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/en/training/custom_diffusion.mdx
View File

@@ -279,9 +279,11 @@ You can also perform inference from one of the complete checkpoint saved during
TODO.
## 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.
More info: https://pytorch.org/docs/stable/generated/torch.optim.Optimizer.zero_grad.html
## Experimental results
You can refer to [our webpage](https://www.cs.cmu.edu/~custom-diffusion/) that discusses our experiments in detail.