Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
mvpatel2000 committed Oct 4, 2024
1 parent 335a0b7 commit f9d4681
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion composer/utils/reproducibility.py
Original file line number Diff line number Diff line change
Expand Up @@ -127,7 +127,7 @@ def configure_deterministic_mode():
# See https://pytorch.org/docs/stable/generated/torch.use_deterministic_algorithms.html
# and https://docs.nvidia.com/cuda/cublas/index.html#cublasApi_reproducibility
os.environ['CUBLAS_WORKSPACE_CONFIG'] = ':4096:8'
log.warning('Deterministic mode is activated. This will negatively impact performance.', category=UserWarning)
log.info('Deterministic mode is activated. This will negatively impact performance.', category=UserWarning)


def get_random_seed() -> int:
Expand Down

0 comments on commit f9d4681

Please sign in to comment.