Compare commits
2 Commits
578f0ceea1
...
ac6400e93a
Author | SHA1 | Date |
---|---|---|
|
ac6400e93a | |
|
0e6a5b8925 |
|
@ -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__':
|
||||
|
|
|
@ -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__':
|
||||
|
|
Binary file not shown.
Binary file not shown.
Loading…
Reference in New Issue