From 4ce43f8c822d36ea88e6d59b221ccb372757f18c Mon Sep 17 00:00:00 2001 From: Guo Liyong Date: Fri, 4 Jun 2021 18:27:27 +0800 Subject: [PATCH] rename exp_dir --- egs/librispeech/asr/simple_v1/mmi_att_transformer_decode.py | 2 +- egs/librispeech/asr/simple_v1/mmi_att_transformer_train.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/egs/librispeech/asr/simple_v1/mmi_att_transformer_decode.py b/egs/librispeech/asr/simple_v1/mmi_att_transformer_decode.py index d6442510..a616bca9 100755 --- a/egs/librispeech/asr/simple_v1/mmi_att_transformer_decode.py +++ b/egs/librispeech/asr/simple_v1/mmi_att_transformer_decode.py @@ -242,7 +242,7 @@ def main(): output_beam_size = args.output_beam_size - exp_dir = Path('exp-' + model_type + '-noam-mmi-att-musan-sa-vgg') + exp_dir = Path('exp-' + model_type + '-mmi-att-sa-vgg-normlayer') setup_logger('{}/log/log-decode'.format(exp_dir), log_level='debug') logging.info(f'output_beam_size: {output_beam_size}') diff --git a/egs/librispeech/asr/simple_v1/mmi_att_transformer_train.py b/egs/librispeech/asr/simple_v1/mmi_att_transformer_train.py index cb1f236e..f1ab15b9 100755 --- a/egs/librispeech/asr/simple_v1/mmi_att_transformer_train.py +++ b/egs/librispeech/asr/simple_v1/mmi_att_transformer_train.py @@ -465,7 +465,7 @@ def run(rank, world_size, args): fix_random_seed(42) setup_dist(rank, world_size, args.master_port) - exp_dir = Path('exp-' + model_type + '-noam-mmi-att-musan-sa-vgg') + exp_dir = Path('exp-' + model_type + '-mmi-att-sa-vgg-normlayer') setup_logger(f'{exp_dir}/log/log-train-{rank}') if args.tensorboard and rank == 0: tb_writer = SummaryWriter(log_dir=f'{exp_dir}/tensorboard') @@ -526,7 +526,8 @@ def run(rank, world_size, args): num_classes=len(phone_ids) + 1, # +1 for the blank symbol subsampling_factor=4, num_decoder_layers=num_decoder_layers, - vgg_frontend=True) + vgg_frontend=True, + is_espnet_structure=True) elif model_type == "contextnet": model = ContextNet( num_features=80,