from local

This commit is contained in:
dohe0342 2023-01-30 16:56:28 +09:00
parent 6f371e84d9
commit 25cf8e9546
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=32)
self.adapters = ResidualAdapterModule(proj_dim=512)
for p in self.adapters.parameters():
p.data /= 10.