Merge pull request #54 from sooftware/tpu

Tpu
sooftware · May 9, 2021 · 8f0fb02 · 8f0fb02
2 parents 99e06e6 + 94b4000
commit 8f0fb02
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 1 deletion.
diff --git a/configs/trainer/gpu.yaml b/configs/trainer/gpu.yaml
@@ -1,6 +1,7 @@
 seed: 1
 accelerator: dp
 precision: 16
+num_workers: 4
 accumulate_grad_batches: 4
 amp_backend: apex
 auto_select_gpus: True

diff --git a/configs/trainer/tpu.yaml b/configs/trainer/tpu.yaml
@@ -1,6 +1,7 @@
 seed: 1
 accelerator: dp
 precision: 16
+num_workers: 4
 accumulate_grad_batches: 4
 amp_backend: apex
 check_val_every_n_epoch: 1

diff --git a/lightning_asr/model/model.py b/lightning_asr/model/model.py
@@ -65,7 +65,6 @@ def __init__(
     ) -> None:
         super(ConformerLSTMModel, self).__init__()
         self.configs = configs
-        self.total_steps = configs.warmup_steps + configs.decay_steps
         self.gradient_clip_val = configs.gradient_clip_val
         self.teacher_forcing_ratio = configs.teacher_forcing_ratio
         self.vocab = vocab