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