from local

This commit is contained in:
dohe0342 2023-01-31 15:50:45 +09:00
parent 39f05501e0
commit 84d4c49b8d
3 changed files with 1 additions and 1 deletions

View File

@ -39,7 +39,7 @@ logger = logging.getLogger().setLevel(logging.INFO)
class TransformerEncoderAdapter(TransformerEncoder):
def __init__(self, args: Wav2Vec2Config):
super().__init__(args)
self.adapters = ResidualAdapterModule(proj_dim=512)
self.adapters = ResidualAdapterModule(proj_dim=384)
for p in self.adapters.parameters():
p.data /= 10.