From 123e55e04a7d508e9f48e57a9ffb5ca27fe0f3b1 Mon Sep 17 00:00:00 2001 From: aditya0by0 Date: Sat, 5 Jul 2025 23:03:18 +0200 Subject: [PATCH] make trainer op deterministic --- chebai/trainer/CustomTrainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/chebai/trainer/CustomTrainer.py b/chebai/trainer/CustomTrainer.py index 874d6b35..2ecee680 100644 --- a/chebai/trainer/CustomTrainer.py +++ b/chebai/trainer/CustomTrainer.py @@ -25,7 +25,7 @@ def __init__(self, *args, **kwargs): """ self.init_args = args self.init_kwargs = kwargs - super().__init__(*args, **kwargs) + super().__init__(*args, **kwargs, deterministic=True) # instantiation custom logger connector self._logger_connector.on_trainer_init(self.logger, 1) # log additional hyperparameters to wandb