Yet another pseudo_huber fix. Still drinking coffee.

2026-04-19 10:05:40 -06:00
parent fc85410c9a
commit 998a02f30e
1 changed files with 1 additions and 1 deletions
--- a/extensions_built_in/sd_trainer/SDTrainer.py
+++ b/extensions_built_in/sd_trainer/SDTrainer.py
@@ -783,7 +783,7 @@ class SDTrainer(BaseSDTrainProcess):
                pred = t0
            if self.train_config.loss_type == "pseudo_huber":
                diff = pred.float() - target.float()
-                c=0.0
+                c=0.01
                loss =(torch.sqrt(diff.pow(2) + c ** 2) - c)
            elif self.train_config.loss_type == "mae":
                loss = torch.nn.functional.l1_loss(pred.float(), target.float(), reduction="none")