Added weighing to DFE

2025-01-22 08:50:57 -07:00
parent 89dd041b97
commit 04abe57c76
2 changed files with 3 additions and 1 deletions
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -403,6 +403,7 @@ class TrainConfig:
        
        # diffusion feature extractor
        self.diffusion_feature_extractor_path = kwargs.get('diffusion_feature_extractor_path', None)
+        self.diffusion_feature_extractor_weight = kwargs.get('diffusion_feature_extractor_weight', 0.1)
        
        # optimal noise pairing
        self.optimal_noise_pairing_samples = kwargs.get('optimal_noise_pairing_samples', 1)