geoffrey-dawson
commited on
changed max epochs to 5
Browse files- config.yaml +1 -1
config.yaml
CHANGED
@@ -17,7 +17,7 @@ trainer:
|
|
17 |
monitor: val/loss
|
18 |
patience: 30
|
19 |
|
20 |
-
max_epochs:
|
21 |
check_val_every_n_epoch: 1
|
22 |
log_every_n_steps: 1
|
23 |
enable_checkpointing: true
|
|
|
17 |
monitor: val/loss
|
18 |
patience: 30
|
19 |
|
20 |
+
max_epochs: 5
|
21 |
check_val_every_n_epoch: 1
|
22 |
log_every_n_steps: 1
|
23 |
enable_checkpointing: true
|