From d95750278b5589cac7b5ab2cd8073cb041746601 Mon Sep 17 00:00:00 2001 From: pkufool Date: Mon, 11 Apr 2022 15:56:45 +0800 Subject: [PATCH] Minor fixes --- .../ASR/pruned_transducer_stateless2/train.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py index 04a9aebb4..d08fa15b5 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py @@ -28,6 +28,18 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3" --exp-dir pruned_transducer_stateless2/exp \ --full-libri 1 \ --max-duration 300 + +# For mix precision training: + +./pruned_transducer_stateless2/train.py \ + --world-size 4 \ + --num-epochs 30 \ + --start-epoch 0 \ + --use_fp16 1 \ + --exp-dir pruned_transducer_stateless2/exp \ + --full-libri 1 \ + --max-duration 550 + """ @@ -710,7 +722,9 @@ def train_one_epoch( ) if tb_writer is not None: - tb_writer.add_scalar("train/learning_rate", cur_params.batch_idx_train) + tb_writer.add_scalar( + "train/learning_rate", cur_lr, params.batch_idx_train + ) loss_info.write_summary( tb_writer, "train/current_", params.batch_idx_train