From 44d75e22c98c4276a494e2b5e6329fc90c36e78e Mon Sep 17 00:00:00 2001 From: yaozengwei Date: Wed, 4 May 2022 21:37:55 +0800 Subject: [PATCH] rename pruned_transducer_stateless3 to pruned_transducer_stateless4 --- .../__init__.py | 0 .../asr_datamodule.py | 0 .../beam_search.py | 0 .../conformer.py | 0 .../decode.py | 0 .../decoder.py | 0 .../encoder_interface.py | 0 .../export.py | 0 .../joiner.py | 0 .../model.py | 0 .../optim.py | 0 .../scaling.py | 0 .../train.py | 6 ++++-- 13 files changed, 4 insertions(+), 2 deletions(-) rename egs/librispeech/ASR/{pruned_transducer_stateless3 => pruned_transducer_stateless4}/__init__.py (100%) rename egs/librispeech/ASR/{pruned_transducer_stateless3 => pruned_transducer_stateless4}/asr_datamodule.py (100%) rename egs/librispeech/ASR/{pruned_transducer_stateless3 => pruned_transducer_stateless4}/beam_search.py (100%) rename egs/librispeech/ASR/{pruned_transducer_stateless3 => pruned_transducer_stateless4}/conformer.py (100%) rename egs/librispeech/ASR/{pruned_transducer_stateless3 => pruned_transducer_stateless4}/decode.py (100%) rename egs/librispeech/ASR/{pruned_transducer_stateless3 => pruned_transducer_stateless4}/decoder.py (100%) rename egs/librispeech/ASR/{pruned_transducer_stateless3 => pruned_transducer_stateless4}/encoder_interface.py (100%) rename egs/librispeech/ASR/{pruned_transducer_stateless3 => pruned_transducer_stateless4}/export.py (100%) rename egs/librispeech/ASR/{pruned_transducer_stateless3 => pruned_transducer_stateless4}/joiner.py (100%) rename egs/librispeech/ASR/{pruned_transducer_stateless3 => pruned_transducer_stateless4}/model.py (100%) rename egs/librispeech/ASR/{pruned_transducer_stateless3 => pruned_transducer_stateless4}/optim.py (100%) rename egs/librispeech/ASR/{pruned_transducer_stateless3 => pruned_transducer_stateless4}/scaling.py (100%) rename egs/librispeech/ASR/{pruned_transducer_stateless3 => pruned_transducer_stateless4}/train.py (99%) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/__init__.py b/egs/librispeech/ASR/pruned_transducer_stateless4/__init__.py similarity index 100% rename from egs/librispeech/ASR/pruned_transducer_stateless3/__init__.py rename to egs/librispeech/ASR/pruned_transducer_stateless4/__init__.py diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/asr_datamodule.py b/egs/librispeech/ASR/pruned_transducer_stateless4/asr_datamodule.py similarity index 100% rename from egs/librispeech/ASR/pruned_transducer_stateless3/asr_datamodule.py rename to egs/librispeech/ASR/pruned_transducer_stateless4/asr_datamodule.py diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/beam_search.py b/egs/librispeech/ASR/pruned_transducer_stateless4/beam_search.py similarity index 100% rename from egs/librispeech/ASR/pruned_transducer_stateless3/beam_search.py rename to egs/librispeech/ASR/pruned_transducer_stateless4/beam_search.py diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless4/conformer.py similarity index 100% rename from egs/librispeech/ASR/pruned_transducer_stateless3/conformer.py rename to egs/librispeech/ASR/pruned_transducer_stateless4/conformer.py diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/decode.py b/egs/librispeech/ASR/pruned_transducer_stateless4/decode.py similarity index 100% rename from egs/librispeech/ASR/pruned_transducer_stateless3/decode.py rename to egs/librispeech/ASR/pruned_transducer_stateless4/decode.py diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/decoder.py b/egs/librispeech/ASR/pruned_transducer_stateless4/decoder.py similarity index 100% rename from egs/librispeech/ASR/pruned_transducer_stateless3/decoder.py rename to egs/librispeech/ASR/pruned_transducer_stateless4/decoder.py diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/encoder_interface.py b/egs/librispeech/ASR/pruned_transducer_stateless4/encoder_interface.py similarity index 100% rename from egs/librispeech/ASR/pruned_transducer_stateless3/encoder_interface.py rename to egs/librispeech/ASR/pruned_transducer_stateless4/encoder_interface.py diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/export.py b/egs/librispeech/ASR/pruned_transducer_stateless4/export.py similarity index 100% rename from egs/librispeech/ASR/pruned_transducer_stateless3/export.py rename to egs/librispeech/ASR/pruned_transducer_stateless4/export.py diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/joiner.py b/egs/librispeech/ASR/pruned_transducer_stateless4/joiner.py similarity index 100% rename from egs/librispeech/ASR/pruned_transducer_stateless3/joiner.py rename to egs/librispeech/ASR/pruned_transducer_stateless4/joiner.py diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/model.py b/egs/librispeech/ASR/pruned_transducer_stateless4/model.py similarity index 100% rename from egs/librispeech/ASR/pruned_transducer_stateless3/model.py rename to egs/librispeech/ASR/pruned_transducer_stateless4/model.py diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/optim.py b/egs/librispeech/ASR/pruned_transducer_stateless4/optim.py similarity index 100% rename from egs/librispeech/ASR/pruned_transducer_stateless3/optim.py rename to egs/librispeech/ASR/pruned_transducer_stateless4/optim.py diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/scaling.py b/egs/librispeech/ASR/pruned_transducer_stateless4/scaling.py similarity index 100% rename from egs/librispeech/ASR/pruned_transducer_stateless3/scaling.py rename to egs/librispeech/ASR/pruned_transducer_stateless4/scaling.py diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/train.py b/egs/librispeech/ASR/pruned_transducer_stateless4/train.py similarity index 99% rename from egs/librispeech/ASR/pruned_transducer_stateless3/train.py rename to egs/librispeech/ASR/pruned_transducer_stateless4/train.py index 83d31c64e..33478c630 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless3/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless4/train.py @@ -74,8 +74,10 @@ from torch.utils.tensorboard import SummaryWriter from icefall import diagnostics from icefall.checkpoint import load_checkpoint, remove_checkpoints from icefall.checkpoint import save_checkpoint as save_checkpoint_impl -from icefall.checkpoint import save_checkpoint_with_global_batch_idx -from icefall.checkpoint import update_averaged_model +from icefall.checkpoint import ( + save_checkpoint_with_global_batch_idx, + update_averaged_model, +) from icefall.dist import cleanup_dist, setup_dist from icefall.env import get_env_info from icefall.utils import AttributeDict, MetricsTracker, setup_logger, str2bool