fix batch_size

Roman Suvorov · Roman Suvorov · commit 5fdb4d58f6f2 · 2020-02-04T13:27:41.000+03:00
diff --git a/task5_text_transformer.ipynb b/task5_text_transformer.ipynb
@@ -1047,7 +1047,7 @@
     "                                         lm_cross_entropy,\n",
     "                                         lr=2e-3,\n",
     "                                         epoch_n=2000,\n",
-    "                                         batch_size=8,\n",
+    "                                         batch_size=512,\n",
     "                                         device='cuda',\n",
     "                                         early_stopping_patience=50,\n",
     "                                         max_batches_per_epoch_train=1000,\n",
