from local

This commit is contained in:
dohe0342 2023-01-03 13:39:04 +09:00
parent df70ae6a5c
commit 5dee34188c
3 changed files with 2 additions and 2 deletions

View File

@ -39,8 +39,8 @@ logger = logging.getLogger().setLevel(logging.INFO)
class TransformerEncoderAdapter(TransformerEncoder):
def __init__(self, args: Wav2Vec2Config):
super().__init__(args)
#self.adapters = ResidualAdapterModule()
self.adapters = ConvolutionModule(768, 31)
self.adapters = ResidualAdapterModule()
#self.adapters = ConvolutionModule(768, 31)
for p in self.adapters.parameters():
#p.data = nn.Parameter(torch.zeros(p.size()).to('cuda'))

Binary file not shown.