Loading the model in mlflow

I am trying to load the model, created and saved during the model training phase in mlflow. However, it says I don’t have “models” directory. I have tried to search on this, but couldn’t identify the issue. Need help!

Ok, this code did the trick for me.
model_load = mlflow.pytorch.load_model(“runs:/e2c089d1e2a44dbca75c25bf82bd7046/models”)

Close this thread please. Thanks!

1 Like