updates the patience
Browse files
train.py
CHANGED
|
@@ -42,14 +42,14 @@ lr_monitor = LearningRateMonitor(logging_interval="step")
|
|
| 42 |
# early stopping
|
| 43 |
early_stopping = EarlyStopping(
|
| 44 |
monitor="val_loss",
|
| 45 |
-
patience=
|
| 46 |
verbose=True,
|
| 47 |
mode="min",
|
| 48 |
)
|
| 49 |
|
| 50 |
# Init trainer
|
| 51 |
trainer = L.Trainer(
|
| 52 |
-
max_epochs=
|
| 53 |
accelerator="auto",
|
| 54 |
devices="auto",
|
| 55 |
logger=logger,
|
|
|
|
| 42 |
# early stopping
|
| 43 |
early_stopping = EarlyStopping(
|
| 44 |
monitor="val_loss",
|
| 45 |
+
patience=10,
|
| 46 |
verbose=True,
|
| 47 |
mode="min",
|
| 48 |
)
|
| 49 |
|
| 50 |
# Init trainer
|
| 51 |
trainer = L.Trainer(
|
| 52 |
+
max_epochs=50,
|
| 53 |
accelerator="auto",
|
| 54 |
devices="auto",
|
| 55 |
logger=logger,
|