HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Update check_version() for inequality support (#4182)

Browse source
This commit is contained in:
Glenn Jocher 2023-08-05 18:23:51 +02:00 • committed by GitHub
parent 038558cfab
commit eb80ba9ce0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 56 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

5
ultralytics/utils/callbacks/mlflow.py
View file

@ -18,7 +18,7 @@ except (ImportError, AssertionError):
def on_pretrain_routine_end(trainer):
"""Logs training parameters to MLflow."""
global mlflow, run, run_id, experiment_name
global mlflow, run, experiment_name
if os.environ.get('MLFLOW_TRACKING_URI') is None:
mlflow = None
@ -39,8 +39,7 @@ def on_pretrain_routine_end(trainer):
run, active_run = mlflow, mlflow.active_run()
if not active_run:
active_run = mlflow.start_run(experiment_id=experiment.experiment_id, run_name=run_name)
run_id = active_run.info.run_id
LOGGER.info(f'{prefix}Using run_id({run_id}) at {mlflow_location}')
LOGGER.info(f'{prefix}Using run_id({active_run.info.run_id}) at {mlflow_location}')
run.log_params(vars(trainer.model.args))
except Exception as err:
LOGGER.error(f'{prefix}Failing init - {repr(err)}')