Example #1
0
    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,
        )
Example #2
0
def test_callback_is_picklable():
    cb = __MlflowFastaiCallback(BatchMetricsLogger(run_id="1234"),
                                log_models=True,
                                is_fine_tune=False)
    pickle.dumps(cb)