def getFastaiCallback(metrics_logger, is_fine_tune=False): from mlflow.fastai.callback import __MlflowFastaiCallback return __MlflowFastaiCallback( metrics_logger=metrics_logger, log_models=log_models, is_fine_tune=is_fine_tune, )
def test_callback_is_picklable(): cb = __MlflowFastaiCallback(BatchMetricsLogger(run_id="1234"), log_models=True, is_fine_tune=False) pickle.dumps(cb)