From fd526254416669ce3caf65ffaa0a7919bf664466 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Thu, 29 Sep 2022 14:30:08 +0530
Subject: [PATCH] add job id to logging

---
 cli/train.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/cli/train.py b/cli/train.py
index 5a056a2..16677b4 100644
--- a/cli/train.py
+++ b/cli/train.py
@@ -1,3 +1,4 @@
+import os
 import hydra
 from hydra.utils import instantiate
 from omegaconf import DictConfig
@@ -12,7 +13,7 @@ def main(config: DictConfig):
     callbacks = []
     callbacks.append(TQDMProgressBar(refresh_rate=10))
     logger = MLFlowLogger(experiment_name=config.mlflow.experiment_name,
-                            run_name=config.mlflow.run_name)
+                            run_name=config.mlflow.run_name, tags={"JOB_ID":os.environ.get("SLURM_JOBID")})
 
 
     parameters = config.hyperparameters