Add mlflow.autolog(log_model_signatures=False) to the second stage of the training task only.
Add mlflow.autolog(log_model_signatures=False) to the second stage of the training task only.