Can't able to load pytorch models in mlflow

1

I have logged my pytorch model in mlflow and now want to load it but it says no such directory found please help!!!

mlflow2
mlflow21771×921 119 KB

2

models path in mlflow

mlflow1
mlflow11460×907 66.6 KB

3

This is how I logged the model

mlflow3
mlflow31909×1022 112 KB

4

I would suggest to post in the mlflow repository, as the issues doesn’t seem to be PyTorch-specific.