Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add job id to logging

This commit is contained in:
shahules786 2022-09-29 14:30:08 +05:30
parent e22cecaf20
commit fd52625441
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cli/train.py
View File

@ -1,3 +1,4 @@
import os
import hydra import hydra
from hydra.utils import instantiate from hydra.utils import instantiate
from omegaconf import DictConfig from omegaconf import DictConfig
@ -12,7 +13,7 @@ def main(config: DictConfig):
callbacks = [] callbacks = []
callbacks.append(TQDMProgressBar(refresh_rate=10)) callbacks.append(TQDMProgressBar(refresh_rate=10))
logger = MLFlowLogger(experiment_name=config.mlflow.experiment_name, 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 parameters = config.hyperparameters