Add model saving for qrun and workflow example

2026-06-06 05:51:17 +08:00 · 2021-02-07 10:06:33 +08:00
parent 07b905c153
commit c948385e76
2 changed files with 2 additions and 0 deletions
--- a/examples/workflow_by_code.py
+++ b/examples/workflow_by_code.py
@@ -112,6 +112,7 @@ if __name__ == "__main__":
    with R.start(experiment_name="workflow"):
        R.log_params(**flatten_dict(task))
        model.fit(dataset)
+        R.save_objects(**{"params.pkl": model})

        # prediction
        recorder = R.get_recorder()