add one more layer to model dir

This commit is contained in:
deng 2023-04-11 11:56:56 +08:00
parent 0e6a5b8925
commit ac6400e93a
2 changed files with 2 additions and 2 deletions

View File

@ -64,7 +64,7 @@ def pull_model(server_uri: str,
print(f'input data: {fake_data}, predictions: {output}')
# save it to local file system
mlflow.pyfunc.save_model(path='./model', python_model=model)
mlflow.pyfunc.save_model(path=f'./model/{exp_name}', python_model=model)
if __name__ == '__main__':

View File

@ -21,7 +21,7 @@ def main():
print(my_fortune)
# save model and env to local file system
mlflow.pytorch.save_model(model, './model')
mlflow.pytorch.save_model(model, './model/fortune_predict_model')
if __name__ == '__main__':