From 837d0ecafaebece46e427c952d805c2235e0e15f Mon Sep 17 00:00:00 2001 From: deepanker13 Date: Fri, 12 Jan 2024 03:02:54 +0530 Subject: [PATCH 1/2] kwarg fix --- examples/sdk/train_api.ipynb | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/sdk/train_api.ipynb b/examples/sdk/train_api.ipynb index 242efe8ebb..80fe0c08d9 100644 --- a/examples/sdk/train_api.ipynb +++ b/examples/sdk/train_api.ipynb @@ -91,6 +91,7 @@ " disable_tqdm=False,\n", " resume_from_checkpoint=True,\n", " remove_unused_columns=True,\n", + " ddp_backend=\"nccl\",\n", " ),\n", " ),\n", " resources_per_worker={\n", @@ -128,7 +129,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.11.6" + "version": "3.11.4" } }, "nbformat": 4, From 153afd9a857d626a58f1cd4cd6013c826543126d Mon Sep 17 00:00:00 2001 From: deepanker13 Date: Fri, 12 Jan 2024 03:10:57 +0530 Subject: [PATCH 2/2] retriggering ci --- examples/sdk/train_api.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/sdk/train_api.ipynb b/examples/sdk/train_api.ipynb index 80fe0c08d9..96c4e34c9f 100644 --- a/examples/sdk/train_api.ipynb +++ b/examples/sdk/train_api.ipynb @@ -91,7 +91,7 @@ " disable_tqdm=False,\n", " resume_from_checkpoint=True,\n", " remove_unused_columns=True,\n", - " ddp_backend=\"nccl\",\n", + " ddp_backend=\"nccl\", # change the backend to gloo if you want cpu based training and remove the gpu key in resources_per_worker\n", " ),\n", " ),\n", " resources_per_worker={\n",