Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "a6b77598805f4e3c24a47767d503dc6ea20d1381"
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:
(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,
)
......
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