cancel
Showing results for 
Search instead for 
Did you mean: 
Machine Learning
Dive into the world of machine learning on the Databricks platform. Explore discussions on algorithms, model training, deployment, and more. Connect with ML enthusiasts and experts.
cancel
Showing results for 
Search instead for 
Did you mean: 

Forum Posts

Itachi_Naruto
by New Contributor II
  • 1423 Views
  • 2 replies
  • 0 kudos

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...

error message
  • 1423 Views
  • 2 replies
  • 0 kudos
Latest Reply
Anonymous
Not applicable
  • 0 kudos

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
1 More Replies
Labels