From a21c5343a5a4f0840fd5adfeaab32c8487379cd5 Mon Sep 17 00:00:00 2001
From: Vladislav Tumko <56307628+vectorvp@users.noreply.github.com>
Date: Fri, 20 Jun 2025 09:03:59 +0400
Subject: [PATCH 1/2] refactor: return custom exp name if is set

For now experiment name returned in automatic way, no matter if is set in config or not.
---
 scripts/f_environment.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scripts/f_environment.py b/scripts/f_environment.py
index b56ce9f..f92d67c 100644
--- a/scripts/f_environment.py
+++ b/scripts/f_environment.py
@@ -156,7 +156,7 @@ def validate_config(config):
     
     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

From f1550c9781a45ec56191a63ba8fc38084fdfa783 Mon Sep 17 00:00:00 2001
From: Vladislav Tumko <56307628+vectorvp@users.noreply.github.com>
Date: Fri, 20 Jun 2025 11:02:31 +0400
Subject: [PATCH 2/2] refactor: return custom exp name if it is set

---
 scripts/f_environment.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/f_environment.py b/scripts/f_environment.py
index f92d67c..5fd0065 100644
--- a/scripts/f_environment.py
+++ b/scripts/f_environment.py
@@ -152,7 +152,7 @@ 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}"