diff --git a/docs/tutorials/custom_segmentation_trainer.ipynb b/docs/tutorials/custom_segmentation_trainer.ipynb
index 9bc287ed01a..9732a50ac8e 100644
--- a/docs/tutorials/custom_segmentation_trainer.ipynb
+++ b/docs/tutorials/custom_segmentation_trainer.ipynb
@@ -299,7 +299,7 @@
     }
    ],
    "source": [
-    "trainer = pl.Trainer(min_epochs=150, max_epochs=300, log_every_n_steps=50)"
+    "trainer = pl.Trainer(min_epochs=150, max_epochs=250, log_every_n_steps=50)"
    ]
   },
   {
diff --git a/docs/tutorials/custom_segmentation_trainer.py b/docs/tutorials/custom_segmentation_trainer.py
index 4b13edae77d..48ed6c7b56a 100644
--- a/docs/tutorials/custom_segmentation_trainer.py
+++ b/docs/tutorials/custom_segmentation_trainer.py
@@ -170,7 +170,7 @@ def on_train_epoch_start(self) -> None:
 # validate that the task's hyperparameters are as expected
 task.hparams
 
-trainer = pl.Trainer(min_epochs=150, max_epochs=300, log_every_n_steps=50)
+trainer = pl.Trainer(min_epochs=150, max_epochs=250, log_every_n_steps=50)
 
 trainer.fit(task, dm)