Skip to content

Commit

Permalink
Expanding comment
Browse files Browse the repository at this point in the history
Expanding comment to create a new cell to log the artifact and register the model after a model run id has been created.
  • Loading branch information
cargecla1 committed Mar 11, 2024
1 parent 22f75f5 commit ee55932
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/userguide/torch_forecasting_models.md
Original file line number Diff line number Diff line change
Expand Up @@ -568,7 +568,7 @@ mlflow.pytorch.get_default_pip_requirements()
# Set tracking uri
mlflow.set_tracking_uri("sqlite:///mlruns.db")

# Save Darts model
# Save Darts model (this need to be added via new cell)
mlflow.log_artifact("NBeatsModel.pickle")

# Registering model
Expand Down

0 comments on commit ee55932

Please sign in to comment.