Commit 2f23437
Add (Scheduled) Pseudo-Huber Loss training scripts to research projects (#7527)
* add scheduled pseudo-huber loss training scripts
See #7488
* add reduction modes to huber loss
* [DB Lora] *2 multiplier to huber loss cause of 1/2 a^2 conv.
pairing of kohya-ss/sd-scripts@c6495de
* [DB Lora] add option for smooth l1 (huber / delta)
Pairing of kohya-ss/sd-scripts@dd22958
* [DB Lora] unify huber scheduling
Pairing of kohya-ss/sd-scripts@19a834c
* [DB Lora] add snr huber scheduler
Pairing of kohya-ss/sd-scripts@47fb1a6
* fixup examples link
* use snr schedule by default in DB
* update all huber scripts with snr
* code quality
* huber: make style && make quality
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>1 parent 2523390 commit 2f23437
File tree
8 files changed
+10106
-0
lines changed- examples/research_projects/scheduled_huber_loss_training
- dreambooth
- text_to_image
8 files changed
+10106
-0
lines changedLines changed: 15 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
0 commit comments