claimed-framework / claimed-framework/iterate_deprecated
Option to track CPU/GPU metrics
Open
- 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
Assessment
This issue has not been assessed yet.