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,
|