changed max epochs to 5 (#7)
Browse files- changed max epochs to 5 (93b3d6298021638af3540b6d9c4cf4f6c7b42f7c)
Co-authored-by: Geoffrey Dawson <[email protected]>
- 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
|