From 176d22c3db8fecc0f15ff0838eef44ff55e56807 Mon Sep 17 00:00:00 2001 From: Caleb Robinson Date: Fri, 23 Feb 2024 17:25:39 +0000 Subject: [PATCH] Update num epochs --- docs/tutorials/custom_segmentation_trainer.ipynb | 2 +- docs/tutorials/custom_segmentation_trainer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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)