From 48c22691e3512eaf0bbab761cda67ce09b6348ea Mon Sep 17 00:00:00 2001 From: ddobokki <44228269+ddobokki@users.noreply.github.com> Date: Fri, 20 May 2022 05:42:44 +0900 Subject: [PATCH] Fix bug in Wav2Vec2 pretrain example (#17326) --- examples/research_projects/wav2vec2/run_pretrain.py | 1 - 1 file changed, 1 deletion(-) diff --git a/examples/research_projects/wav2vec2/run_pretrain.py b/examples/research_projects/wav2vec2/run_pretrain.py index cd35caaaa83..fb430d14074 100755 --- a/examples/research_projects/wav2vec2/run_pretrain.py +++ b/examples/research_projects/wav2vec2/run_pretrain.py @@ -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, )