Skip to content

Commit eb1151b

Browse files
Apply suggestions from code review
Co-authored-by: Justus Schock <[email protected]>
1 parent 30bd103 commit eb1151b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pytorch_lightning/trainer/connectors/accelerator_connector.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,8 @@ def _init_deterministic(self, deterministic: bool) -> None:
211211
if deterministic:
212212
# fixing non-deterministic part of horovod
213213
# https://github.com/PyTorchLightning/pytorch-lightning/pull/1572/files#r420279383
214-
os.environ["HOROVOD_FUSION_THRESHOLD"] = str(0)
214+
os.environ["HOROVOD_FUSION_THRESHOLD"] = "0"
215+
215216
# https://docs.nvidia.com/cuda/cublas/index.html#cublasApi_reproducibility
216217
os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8"
217218

0 commit comments

Comments
 (0)