from local
This commit is contained in:
parent
df70ae6a5c
commit
5dee34188c
Binary file not shown.
@ -39,8 +39,8 @@ logger = logging.getLogger().setLevel(logging.INFO)
|
|||||||
class TransformerEncoderAdapter(TransformerEncoder):
|
class TransformerEncoderAdapter(TransformerEncoder):
|
||||||
def __init__(self, args: Wav2Vec2Config):
|
def __init__(self, args: Wav2Vec2Config):
|
||||||
super().__init__(args)
|
super().__init__(args)
|
||||||
#self.adapters = ResidualAdapterModule()
|
self.adapters = ResidualAdapterModule()
|
||||||
self.adapters = ConvolutionModule(768, 31)
|
#self.adapters = ConvolutionModule(768, 31)
|
||||||
|
|
||||||
for p in self.adapters.parameters():
|
for p in self.adapters.parameters():
|
||||||
#p.data = nn.Parameter(torch.zeros(p.size()).to('cuda'))
|
#p.data = nn.Parameter(torch.zeros(p.size()).to('cuda'))
|
||||||
|
|||||||
Binary file not shown.
Reference in New Issue
Block a user