from local

This commit is contained in:
dohe0342 2023-01-30 16:46:59 +09:00
parent 7bc14ba82f
commit 6bdd41f03d
2 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()
self.adapters = ResidualAdapterModule(proj_dim=32)
for p in self.adapters.parameters():
p.data /= 10.