diff --git a/scripts/f_environment.py b/scripts/f_environment.py index b56ce9f..5fd0065 100644 --- a/scripts/f_environment.py +++ b/scripts/f_environment.py @@ -152,11 +152,11 @@ def validate_config(config): else: experiment_name += f"_sd{config.SEED}" - return experiment_name + return experiment_name if not config.EXPERIMENT_NAME else config.EXPERIMENT_NAME experiment_name += f"_sd{config.SEED}" - return experiment_name + return experiment_name if not config.EXPERIMENT_NAME else config.EXPERIMENT_NAME def find_seed_in_weight(weight_name): match = re.search(r'sd(\d+)', weight_name) @@ -173,4 +173,4 @@ def set_deterministic_behaviour(seed): torch.cuda.manual_seed(seed) # Set CUDA seed torch.backends.cudnn.benchmark = False # Disable dynamic tuning torch.use_deterministic_algorithms(True) # Force deterministic behavior - os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8" # CUDA workspace config \ No newline at end of file + os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8" # CUDA workspace config