Fix bug in Wav2Vec2 pretrain example (huggingface#17326)

ddobokki · elusenji · commit da211f8b0058 · 2022-06-12T11:14:34.000+02:00
diff --git a/examples/research_projects/wav2vec2/run_pretrain.py b/examples/research_projects/wav2vec2/run_pretrain.py
@@ -202,7 +202,6 @@ def __call__(self, features: List[Dict[str, Union[List[int], torch.Tensor]]]) ->
             (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,
         )

Original file line number	Diff line number	Diff line change
`@@ -202,7 +202,6 @@ def __call__(self, features: List[Dict[str, Union[List[int], torch.Tensor]]]) ->`
`202`	`202`	`(batch_size, mask_indices_seq_length),`
`203`	`203`	`self.model.config.mask_time_prob,`
`204`	`204`	`self.model.config.mask_time_length,`
`205`		`- device=batch["input_values"].device,`
`206`	`205`	`attention_mask=attention_mask,`
`207`	`206`	`min_masks=2,`
`208`	`207`	`)`