claimed-framework / claimed-framework/iterate_deprecated

Option to track CPU/GPU metrics

Open
#30 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
34
Forks
5
PR merge metrics
No merged PRs in 30d

Description

MLFlow can track the compute metrics similar to WandB. It would be good to give the user the possibility to activate it.

See https://mlflow.org/docs/latest/system-metrics/index.html

The metrics can be logged with:
```
logger = MLFlowLogger(
experiment_name=LOG_EXPERIMENT_NAME,
run_name=run_name,
tracking_uri=f"file:{LOG_DIR}{LOG_PROJECT}",
)
logger.log_hyperparams(config)

# Start logging system metrics
mlflow.start_run(
run_id=logger.run_id,
experiment_id=logger.experiment_id,
log_system_metrics=True)
```

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.