Spaces:
Runtime error
Runtime error
mattricesound
commited on
Commit
•
020bd46
1
Parent(s):
ea00224
Fix config copy-paste bug
Browse files- cfg/config.yaml +2 -2
cfg/config.yaml
CHANGED
@@ -116,11 +116,11 @@ trainer:
|
|
116 |
_target_: pytorch_lightning.Trainer
|
117 |
precision: 32 # Precision used for tensors, default `32`
|
118 |
min_epochs: 0
|
119 |
-
|
120 |
enable_model_summary: False
|
121 |
log_every_n_steps: 1 # Logs metrics every N batches
|
122 |
accumulate_grad_batches: 1
|
123 |
accelerator: ${accelerator}
|
124 |
devices: 1
|
125 |
gradient_clip_val: 10.0
|
126 |
-
|
|
|
116 |
_target_: pytorch_lightning.Trainer
|
117 |
precision: 32 # Precision used for tensors, default `32`
|
118 |
min_epochs: 0
|
119 |
+
max_epochs: -1
|
120 |
enable_model_summary: False
|
121 |
log_every_n_steps: 1 # Logs metrics every N batches
|
122 |
accumulate_grad_batches: 1
|
123 |
accelerator: ${accelerator}
|
124 |
devices: 1
|
125 |
gradient_clip_val: 10.0
|
126 |
+
max_steps: 50000
|