你好,
我有我推入dbfs PyTorch模型現在我想使用MLflow服務模型。我發現模型需要在python_function模型。
我做了下麵的方法
1。負荷模型使用火炬從dbfs負荷的選擇
2。然後保存使用pyfunc python_function模型的模型。save_model函數
3所示。當我注冊模型後得到一個解碼錯誤
我不訓練磚中的任何模型。
進口mlflow mlflow進口。從火炬進口負載pyfunc torch_load py_model = torch_load (“/ dbfs / FileStore /毫升/ ner_model”, map_location = torch_device (ner_gpu_device)) mlflow.pytorch.save_model (py_model、路徑= " / dbfs / FileStore pyfunc / ner_model”)模型= mlflow.pyfunc.load_model (“/ dbfs / FileStore / pyfunc / ner_model”) mlflow.register_model(模型、“ner_model”) #加載python_function模型注冊模型= mlflow.pyfunc.load_model (“/ dbfs / FileStore / pyfunc / ner_model”) model_version = mlflow.register_model(模型、“ner_model”)