mattricesound commited on
Commit
020bd46
1 Parent(s): ea00224

Fix config copy-paste bug

Browse files
Files changed (1) hide show
  1. 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
- num_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
- num_steps: 50000
 
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