from local

This commit is contained in:
dohe0342 2023-01-30 17:02:12 +09:00
parent 46be3a7375
commit 3aaa59a2dd
4 changed files with 2 additions and 2 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=32)
for p in self.adapters.parameters():
p.data /= 10.

View File

@ -58,7 +58,7 @@ else
--enable-spec-aug False \
--multi-optim False \
--world-size 4 \
--num-epochs 200 \
--num-epochs 100 \
--full-libri 1 \
--exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \
--max-duration 150 \