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