accelerate
f648feba - Add `log_artifact`, `log_artifacts` and `log_figure` capabilities to the MLflowTracker. (#3419)

Commit
1 year ago
Add `log_artifact`, `log_artifacts` and `log_figure` capabilities to the MLflowTracker. (#3419) * Added artifacts and figure tracking at MLFlow tracker * Added `log_artifact` to the MLFlowTracker * Remove changes * Added artifacts, artifacts and figure tracking at MLFlow tracker * Improved the docstring * added require_mlflow function at test_utils * add test for MLflowTracker * Bit of litting * Refactor to a more robust test * Revised the test asserts to something more robust. * Removed incorrect import and some litting. * removed commented code * initiate tracker using Accelerator * Added mlflow and matplotlib to setup.py. Guarded and decoredated the functions that required them. * Guarded mlflow import * added matplotlib required warning. * ran style and quality
Author
Parents
Loading