Skip to content

Commit 1a13718

Browse files
authored
Merge pull request #101 from ChEB-AI/fix/trainer_deterministic
Make training deterministic
2 parents a7c838e + 123e55e commit 1a13718

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

chebai/trainer/CustomTrainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def __init__(self, *args, **kwargs):
2525
"""
2626
self.init_args = args
2727
self.init_kwargs = kwargs
28-
super().__init__(*args, **kwargs)
28+
super().__init__(*args, **kwargs, deterministic=True)
2929
# instantiation custom logger connector
3030
self._logger_connector.on_trainer_init(self.logger, 1)
3131
# log additional hyperparameters to wandb

0 commit comments

Comments
 (0)