Skip to content

GitLab

Unverified Commit 48c22691 authored by ddobokki's avatar ddobokki Committed by GitHub
Browse files

Fix bug in Wav2Vec2 pretrain example (#17326)

parent 5d6feecf
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
examples/research_projects/wav2vec2/run_pretrain.py
View file @ 48c22691
...@@ -202,7 +202,6 @@ class DataCollatorForWav2Vec2Pretraining: ...@@ -202,7 +202,6 @@ class DataCollatorForWav2Vec2Pretraining:
(batch_size, mask_indices_seq_length), (batch_size, mask_indices_seq_length),
self.model.config.mask_time_prob, self.model.config.mask_time_prob,
self.model.config.mask_time_length, self.model.config.mask_time_length,
device=batch["input_values"].device,
attention_mask=attention_mask, attention_mask=attention_mask,
min_masks=2, min_masks=2,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment