riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Show Stats 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),
self.model.config.mask_time_prob,
self.model.config.mask_time_length,
device=batch["input_values"].device,
attention_mask=attention_mask,
min_masks=2,
)