Skip to content

GitLab

"research/neural_gpu/data_utils.py" did not exist on "90d6e3b97b6eaa62ddc5f0e8e8173b644de03396"
Switch branch/tag
Find file BlameHistoryPermalink
  • Gleb Akhmerov's avatar
    Dreambooth: reduce VRAM usage (#2039) · a66f2bae
    Gleb Akhmerov authored 
    * Dreambooth: use `optimizer.zero_grad(set_to_none=True)` to reduce VRAM usage

* Allow the user to control `optimizer.zero_grad(set_to_none=True)` with --set_grads_to_none

* Update Dreambooth readme

* Fix link in readme

* Fix header size in readme
    a66f2bae
train_dreambooth.py 35.2 KB