How to Register a ML model using MLflow
Hi,I have a PyTorch model which I have pushed into the dbfs now I want to serve the model using MLflow. I saw that the model needs to be in python_function model.To do that I did the following methods1. load the model from dbfs using torch load optio...
- 2168 Views
- 1 replies
- 0 kudos
Latest Reply
I think you want to use mflow to load the model not pytorch. There is a function in mlflow to load pytorch models https://www.mlflow.org/docs/latest/python_api/mlflow.pytorch.html#mlflow.pytorch.load_modelThen once it's loaded, you can log it and re...
- 0 kudos