Update num epochs

microsoft · Feb 23, 2024 · 176d22c · 176d22c
1 parent f07b4a7
commit 176d22c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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)