Change img multiplier math

2024-07-30 11:33:41 -06:00
parent 443c996e7f
commit 47744373f2
4 changed files with 18 additions and 3 deletions
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -354,6 +354,7 @@ class TrainConfig:

        # adds an additional loss to the network to encourage it output a normalized standard deviation
        self.target_norm_std = kwargs.get('target_norm_std', None)
+        self.target_norm_std_value = kwargs.get('target_norm_std_value', 1.0)


 class ModelConfig: