Added a new experimental linear weighing technique

2024-09-02 09:22:13 -06:00
parent 7d9ab22405
commit d44d4eb61a
4 changed files with 19 additions and 5 deletions
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -359,6 +359,7 @@ class TrainConfig:
        self.target_norm_std = kwargs.get('target_norm_std', None)
        self.target_norm_std_value = kwargs.get('target_norm_std_value', 1.0)
        self.linear_timesteps = kwargs.get('linear_timesteps', False)
+        self.linear_timesteps2 = kwargs.get('linear_timesteps2', False)
        self.disable_sampling = kwargs.get('disable_sampling', False)