[Python] #3 QA 튜토리얼 보완

ratsgo · Mar 6, 2021 · e31d213 · e31d213
1 parent 2493bad
commit e31d213
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/question_answering/train_colab.ipynb b/examples/question_answering/train_colab.ipynb
@@ -84,7 +84,7 @@
         "    doc_stride=64,\n",
         "    do_eval=True,\n",
         "    batch_size=32,\n",
-        "    epochs=10,\n",
+        "    epochs=5,\n",
         ")"
       ],
       "execution_count": null,

diff --git a/examples/question_answering/train_local.py b/examples/question_answering/train_local.py
@@ -20,7 +20,7 @@
             max_query_length=32,
             doc_stride=64,
             batch_size=32,
-            epochs=10,
+            epochs=5,
         )
     # case2 : python train_local.py train_config.json
     elif len(sys.argv) == 2 and sys.argv[-1].endswith(".json"):