Skip to content

Commit 123e55e

Browse files
committed
make trainer op deterministic
1 parent 2ca72e9 commit 123e55e

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)