Bug fix with dataloader. Added a flag to completly disable sampling

2024-08-12 09:19:40 -06:00
parent 89d61a3b8e
commit af108bb964
3 changed files with 11 additions and 4 deletions
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -357,6 +357,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.disable_sampling = kwargs.get('disable_sampling', False)


 class ModelConfig: