bhimrazy commited on
Commit
0e288a7
1 Parent(s): d8488fe

updates the patience

Browse files
Files changed (1) hide show
  1. train.py +2 -2
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=5,
46
  verbose=True,
47
  mode="min",
48
  )
49
 
50
  # Init trainer
51
  trainer = L.Trainer(
52
- max_epochs=20,
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,