Fix bug in Wav2Vec2 pretrain example (#17326)

2022-05-20 05:42:44 +09:00
parent 5d6feecf16
commit 48c22691e3
1 changed files with 0 additions and 1 deletions
--- a/examples/research_projects/wav2vec2/run_pretrain.py
+++ b/examples/research_projects/wav2vec2/run_pretrain.py
@@ -202,7 +202,6 @@ class DataCollatorForWav2Vec2Pretraining:
            (batch_size, mask_indices_seq_length),
            self.model.config.mask_time_prob,
            self.model.config.mask_time_length,
-            device=batch["input_values"].device,
            attention_mask=attention_mask,
            min_masks=2,
        )