diff --git a/stage_one.py b/stage_one.py index 7c82e6706ade0b7bb614bb662df4070a25c0ab54..800be3c4961cd7c20cfa59d33bc4769baa1307d8 100644 --- a/stage_one.py +++ b/stage_one.py @@ -2,7 +2,7 @@ from clearml import Task, Dataset task = Task.init(project_name="pipeline", task_name="pipeline step 1 dataset artifact") -task.add_requirements("./requirements.txt") +task.add_requirements("-r ./requirements.txt") # only create the task, it will be executed remotely later task.execute_remotely() diff --git a/stage_two.py b/stage_two.py index 288f1f00b5124866ce1a00aa3e0b10252214b20d..9c1694c6db053ca9a584b012681435e3fb75c7d7 100644 --- a/stage_two.py +++ b/stage_two.py @@ -77,7 +77,7 @@ def validate(model, testloader, criterion): task = Task.init(project_name="pipeline", task_name="pipeline step 2 train model") -task.add_requirements("./requirements.txt") +task.add_requirements("-r ./requirements.txt") # only create the task, we will actually execute it later task.execute_remotely()