from local

This commit is contained in:
dohe0342 2022-12-10 16:16:50 +09:00
parent 573a79b08d
commit 411c531925
4 changed files with 2 additions and 1 deletions

Binary file not shown.

View File

@ -1231,6 +1231,7 @@ def run(rank, world_size, args, wb=None):
if rank == 0:
print(enc_names)
optimizer_enc = ScaledAdam(
enc_param,
lr=params.peak_enc_lr,

View File

@ -13,7 +13,7 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
--full-libri 0 \
--exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \
--max-duration 250 \
--freeze-finetune-updates 500 \
--freeze-finetune-updates 200 \
--use-fp16 1 \
--peak-enc-lr 0.0001 \
--peak-dec-lr 0.003 \