SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix bug in Wav2Vec2 pretrain example (#17326)

Browse Source
This commit is contained in:
ddobokki
2022-05-20 05:42:44 +09:00
committed by GitHub
parent 5d6feecf16
commit 48c22691e3
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/research_projects/wav2vec2/run_pretrain.py
View File

@@ -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,
) )