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

[Wav2Vec2 and Co] Update init tests for PT 2.1 (#26494)

Browse Source
This commit is contained in:
Sanchit Gandhi
2023-10-03 09:52:34 +01:00
committed by GitHub
parent b5ca8fcd20
commit 768aa3d9cd
5 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/hubert/test_modeling_hubert.py
View File

@@ -419,6 +419,7 @@ class HubertModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
for name, param in model.named_parameters(): for name, param in model.named_parameters():
uniform_init_parms = [ uniform_init_parms = [
"conv.weight", "conv.weight",
"conv.parametrizations.weight",
"masked_spec_embed", "masked_spec_embed",
"quantizer.weight_proj.weight", "quantizer.weight_proj.weight",
] ]
@@ -680,6 +681,7 @@ class HubertRobustModelTest(ModelTesterMixin, unittest.TestCase):
for name, param in model.named_parameters(): for name, param in model.named_parameters():
uniform_init_parms = [ uniform_init_parms = [
"conv.weight", "conv.weight",
"conv.parametrizations.weight",
"masked_spec_embed", "masked_spec_embed",
"quantizer.weight_proj.weight", "quantizer.weight_proj.weight",
] ]

1
tests/models/unispeech/test_modeling_unispeech.py
View File

@@ -421,6 +421,7 @@ class UniSpeechRobustModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.T
for name, param in model.named_parameters(): for name, param in model.named_parameters():
uniform_init_parms = [ uniform_init_parms = [
"conv.weight", "conv.weight",
"conv.parametrizations.weight",
"masked_spec_embed", "masked_spec_embed",
"codevectors", "codevectors",
"quantizer.weight_proj.weight", "quantizer.weight_proj.weight",

2
tests/models/unispeech_sat/test_modeling_unispeech_sat.py
View File

@@ -471,6 +471,7 @@ class UniSpeechSatModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Test
for name, param in model.named_parameters(): for name, param in model.named_parameters():
uniform_init_parms = [ uniform_init_parms = [
"conv.weight", "conv.weight",
"conv.parametrizations.weight",
"masked_spec_embed", "masked_spec_embed",
"codevectors", "codevectors",
"quantizer.weight_proj.weight", "quantizer.weight_proj.weight",
@@ -682,6 +683,7 @@ class UniSpeechSatRobustModelTest(ModelTesterMixin, unittest.TestCase):
for name, param in model.named_parameters(): for name, param in model.named_parameters():
uniform_init_parms = [ uniform_init_parms = [
"conv.weight", "conv.weight",
"conv.parametrizations.weight",
"masked_spec_embed", "masked_spec_embed",
"codevectors", "codevectors",
"quantizer.weight_proj.weight", "quantizer.weight_proj.weight",

1
tests/models/wav2vec2/test_modeling_wav2vec2.py
View File

@@ -625,6 +625,7 @@ class Wav2Vec2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase
for name, param in model.named_parameters(): for name, param in model.named_parameters():
uniform_init_parms = [ uniform_init_parms = [
"conv.weight", "conv.weight",
"conv.parametrizations.weight",
"masked_spec_embed", "masked_spec_embed",
"codevectors", "codevectors",
"quantizer.weight_proj.weight", "quantizer.weight_proj.weight",

1
tests/models/wavlm/test_modeling_wavlm.py
View File

@@ -424,6 +424,7 @@ class WavLMModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
for name, param in model.named_parameters(): for name, param in model.named_parameters():
uniform_init_parms = [ uniform_init_parms = [
"conv.weight", "conv.weight",
"conv.parametrizations.weight",
"masked_spec_embed", "masked_spec_embed",
"codevectors", "codevectors",
"quantizer.weight_proj.weight", "quantizer.weight_proj.weight",