Merge pull request #389 from MrCsabaToth/patch-1

Adding prints, fixes #388
GoogleCloudPlatform · Oct 25, 2023 · 87fc8af · 87fc8af
2 parents 7de35c3 + f1513d9
commit 87fc8af
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/notebooks/text_models/labs/text_generation_using_transformers.ipynb b/notebooks/text_models/labs/text_generation_using_transformers.ipynb
@@ -422,9 +422,11 @@
    "outputs": [],
    "source": [
     "EPOCHS = 1  # increase the number of epochs for better results\n",
+    "print(\"Training started, this could take 4-10 mins per epoch with a T4 GPU...\")\n",
     "model.fit(\n",
     "    # TODO: Fill out the arguments\n",
-    ")"
+    ")\n",
+    "print(\"Training is complete!!\")"
    ]
   },
   {

diff --git a/notebooks/text_models/solutions/text_generation_using_transformers.ipynb b/notebooks/text_models/solutions/text_generation_using_transformers.ipynb
@@ -401,7 +401,9 @@
    "outputs": [],
    "source": [
     "EPOCHS = 1  # increase the number of epochs for better results\n",
-    "model.fit(train_ds, validation_data=val_ds, verbose=2, epochs=EPOCHS)"
+    "print(\"Training started, this could take 4-10 mins per epoch with a T4 GPU...\")\n",
+    "model.fit(train_ds, validation_data=val_ds, verbose=2, epochs=EPOCHS)\n",
+    "print(\"Training is complete!!\")"
    ]
   },
   {