Merge pull request #468 from marrlab/xd_benchmark_fix_model_name

add slurm_id to model name as suffix Update exp_utils.py
marrlab · Sep 28, 2023 · 3479847 · 3479847
2 parents e47e39e + 93122b7
commit 3479847
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/domainlab/compos/exp/exp_utils.py b/domainlab/compos/exp/exp_utils.py
@@ -63,6 +63,9 @@ def mk_model_na(self, tag=None, dd_cut=19):
         model_name = "_".join(list4mname)
         if self.host.args.debug:
             model_name = "debug_" + model_name
+        slurm = os.environ.get('SLURM_JOB_ID')
+        if slurm:
+            model_name = model_name + '_' + slurm
         logger = Logger.get_logger()
         logger.info(f"model name: {model_name}")
         return model_name