Loading the model in mlflow

[mishramarnath]

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!

[mishramarnath]

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

Close this thread please. Thanks!