|
26 | 26 | from typing_extensions import Literal |
27 | 27 |
|
28 | 28 | _log: logging.Logger = logging.getLogger(__name__) |
29 | | -_log.setLevel(logging.DEBUG) # Set logger level to DEBUG to see all messages |
30 | 29 |
|
31 | 30 |
|
32 | 31 | def _check_dist_env() -> bool: |
@@ -136,23 +135,23 @@ def seed(seed: int, deterministic: Optional[Union[str, int]] = None) -> None: |
136 | 135 | raise ValueError( |
137 | 136 | f"Invalid seed value provided: {seed}. Value must be in the range [{min_val}, {max_val}]" |
138 | 137 | ) |
139 | | - _log.debug(f"Setting seed to {seed}") |
| 138 | + _log.info(f"Setting seed to {seed}") |
140 | 139 |
|
141 | 140 | torch.manual_seed(seed) |
142 | 141 | np.random.seed(seed) |
143 | 142 | random.seed(seed) |
144 | 143 | os.environ["PYTHONHASHSEED"] = str(seed) |
145 | 144 |
|
146 | 145 | if deterministic is not None: |
147 | | - _log.debug(f"Setting deterministic debug mode to {deterministic}") |
| 146 | + _log.info(f"Setting deterministic debug mode to {deterministic}") |
148 | 147 | torch.set_deterministic_debug_mode(deterministic) |
149 | 148 | deterministic_debug_mode = torch.get_deterministic_debug_mode() |
150 | 149 | if deterministic_debug_mode == 0: |
151 | | - _log.debug("Disabling cuDNN deterministic mode") |
| 150 | + _log.info("Disabling cuDNN deterministic mode") |
152 | 151 | torch.backends.cudnn.deterministic = False |
153 | 152 | torch.backends.cudnn.benchmark = True |
154 | 153 | else: |
155 | | - _log.debug("Enabling cuDNN deterministic mode") |
| 154 | + _log.info("Enabling cuDNN deterministic mode") |
156 | 155 | torch.backends.cudnn.deterministic = True |
157 | 156 | torch.backends.cudnn.benchmark = False |
158 | 157 | # reference: https://docs.nvidia.com/cuda/cublas/index.html#cublasApi_reproducibility |
|
0 commit comments