update train.py (#12)

tybritten · web-flow · commit 0f7cd50f920f · 2023-10-19T09:10:09.000-04:00
diff --git a/examples/training_container/train.py b/examples/training_container/train.py
@@ -299,7 +299,7 @@ def main():
     )
     client = create_client()
     model = get_or_create_model(client, args.model, pipeline, args.repo)
-    exp = run_experiment(client, config, workdir, model)
+    exp = run_experiment(client, config, workdir, model, args.incremental)
 
     if exp is None:
         print("Aborting pipeline as experiment did not succeed")

Original file line number	Diff line number	Diff line change
`@@ -299,7 +299,7 @@ def main():`
`299`	`299`	`)`
`300`	`300`	`client = create_client()`
`301`	`301`	`model = get_or_create_model(client, args.model, pipeline, args.repo)`
`302`		`- exp = run_experiment(client, config, workdir, model)`
	`302`	`+ exp = run_experiment(client, config, workdir, model, args.incremental)`
`303`	`303`
`304`	`304`	`if exp is None:`
`305`	`305`	`print("Aborting pipeline as experiment did not succeed")`