diff --git a/notebooks/text_models/labs/text_generation_using_transformers.ipynb b/notebooks/text_models/labs/text_generation_using_transformers.ipynb
index 5264d493..4b5eb917 100644
--- 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
index 4758be18..7de985c7 100644
--- 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!!\")"
    ]
   },
   {