Spaces:
Sleeping
Sleeping
File size: 385 Bytes
739dca3 c303abd 5e55742 efc36ce c303abd 5e55742 c303abd efc36ce c303abd efc36ce | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | from config.settings import AppConfig
import mlflow
config = AppConfig()
def log_config():
mlflow.log_param("Detector", config.yolo.model_name)
mlflow.log_param("Safety Model", config.security_detector.model_name)
mlflow.log_param("Depth Model", config.depth.model_name)
def log_metrics(metrics: dict):
for k, v in metrics.items():
mlflow.log_metric(k, v)
|