from local

This commit is contained in:
dohe0342 2023-04-10 18:17:41 +09:00
parent 90f1d6f795
commit 408b4a6f2c
6 changed files with 3 additions and 4 deletions

Binary file not shown.

Binary file not shown.

View File

@ -739,7 +739,7 @@ def load_checkpoint_if_available(
if "cur_batch_idx" in saved_params:
params["cur_batch_idx"] = saved_params["cur_batch_idx"]
#params.batch_idx_train = 0
params.batch_idx_train = 0
return saved_params

View File

@ -53,10 +53,9 @@ if [ $workstation = "whale" ]; then
# --prune-range 10
else
export CUDA_VISIBLE_DEVICES="0,1,2,3"
rm ./pruned_transducer_stateless_d2v_v2/$1/.train.done
#rm ./pruned_transducer_stateless_d2v_v2/$1/.train.done
if [ ! -e ./pruned_transducer_stateless_d2v_v2/$1/.train.done ]; then
./pruned_transducer_stateless_d2v_v2/train_adapter.py \
--start-batch 200 \
--num-buckets 2 \
--add-adapter True \
--adapter-lr 0.02 \
@ -67,7 +66,7 @@ else
--multi-optim False \
--world-size 4 \
--num-epochs 10000 \
--num-updates 301 \
--num-updates 201 \
--save-every-n 100 \
--full-libri 1 \
--exp-dir ./pruned_transducer_stateless_d2v_v2/$1 \