Update Question_Answering.ipynb (NVIDIA#5049)

jubick1337 · Oct 3, 2022 · 32e9966 · 32e9966
1 parent 903b40a
commit 32e9966
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tutorials/nlp/Question_Answering.ipynb b/tutorials/nlp/Question_Answering.ipynb
@@ -354,7 +354,8 @@
         "config.trainer.max_steps = -1 # takes precedence over max_epochs\n",
         "config.trainer.precision = 16\n",
         "config.trainer.devices = [0] # 0 for CPU, or list of the GPUs to use e.g. [0, 1] or [0]\n",
-        "config.trainer.accelerator = \"gpu\""
+        "config.trainer.accelerator = \"gpu\"\n",
+        "config.trainer.strategy=\"ddp_fork\""
       ]
     },
     {