accelerate
Add `log_artifact`, `log_artifacts` and `log_figure` capabilities to the MLflowTracker.
#3419
Merged
Merged
Add `log_artifact`, `log_artifacts` and `log_figure` capabilities to the MLflowTracker. #3419
SunMarc
merged
17 commits into
huggingface:main
from
luiz0992:luiz0992-/--Add-more-tracking-capabilities-to-MLFlowTracker
Added artifacts and figure tracking at MLFlow tracker
d42384dd
Added `log_artifact` to the MLFlowTracker
5d051b9c
Remove changes
748a5877
Added artifacts, artifacts and figure tracking at MLFlow tracker
b404fe90
Improved the docstring
c757df04
SunMarc
commented
on 2025-03-04
added require_mlflow function at test_utils
4cbbe3b5
add test for MLflowTracker
24414989
Bit of litting
4381f451
Refactor to a more robust test
7ad3b041
Revised the test asserts to something more robust.
28f4f401
Removed incorrect import and some litting.
09eadc73
removed commented code
d39a9b19
SunMarc
commented
on 2025-03-04
initiate tracker using Accelerator
a7bdeef7
SunMarc
approved these changes
on 2025-03-05
Added mlflow and matplotlib to setup.py. Guarded and decoredated the …
a5e3d8fe
Guarded mlflow import
5cb9a77f
added matplotlib required warning.
1fc10f28
muellerzr
approved these changes
on 2025-03-12
ran style and quality
05932b81
luiz0992
deleted the luiz0992-/--Add-more-tracking-capabilities-to-MLFlowTracker branch 308 days ago
Login to write a write a comment.
Login via GitHub