Skip to content
Snippets Groups Projects

Always use the same dropout scheduler function

Merged Yoann Schneider requested to merge always-use-dropout-scheduler into main
All threads resolved!
Files
4
+ 1
4
@@ -196,10 +196,7 @@ class GenericTrainingManager:
)
# Handle curriculum dropout
if "dropout_scheduler" in self.params["model_params"]:
func = self.params["model_params"]["dropout_scheduler"]["function"]
T = self.params["model_params"]["dropout_scheduler"]["T"]
self.dropout_scheduler = DropoutScheduler(self.models, func, T)
self.dropout_scheduler = DropoutScheduler(self.models)
self.scaler = GradScaler(enabled=self.params["training_params"]["use_amp"])
Loading