github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Yet another pseudo_huber fix. Still drinking coffee.

Browse Source
This commit is contained in:
Jaret Burkett
2026-04-19 10:05:40 -06:00
parent fc85410c9a
commit 998a02f30e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions_built_in/sd_trainer/SDTrainer.py
View File

@@ -783,7 +783,7 @@ class SDTrainer(BaseSDTrainProcess):
pred = t0 pred = t0
if self.train_config.loss_type == "pseudo_huber": if self.train_config.loss_type == "pseudo_huber":
diff = pred.float() - target.float() diff = pred.float() - target.float()
c=0.0 c=0.01
loss =(torch.sqrt(diff.pow(2) + c ** 2) - c) loss =(torch.sqrt(diff.pow(2) + c ** 2) - c)
elif self.train_config.loss_type == "mae": elif self.train_config.loss_type == "mae":
loss = torch.nn.functional.l1_loss(pred.float(), target.float(), reduction="none") loss = torch.nn.functional.l1_loss(pred.float(), target.float(), reduction="none")