Update trainer.py

jykang0332 · Apr 19, 2023 · 09136bd · 09136bd
1 parent 6869fa3
commit 09136bd
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/valle/bin/trainer.py b/valle/bin/trainer.py
@@ -521,7 +521,6 @@ def compute_loss(
  y=audio_features,
  y_lens=audio_features_lens,
  train_stage=params.train_stage,
- batch_idx_train=params.batch_idx_train,
  )
 
  assert loss.requires_grad == is_training