utils.callbacks.mlflow_

utils.callbacks.mlflow_

MLFlow module for trainer callbacks

Classes

Name Description
SaveAxolotlConfigtoMlflowCallback Callback to save axolotl config to mlflow

SaveAxolotlConfigtoMlflowCallback

utils.callbacks.mlflow_.SaveAxolotlConfigtoMlflowCallback(
    self,
    axolotl_config_path,
)

Callback to save axolotl config to mlflow