From d6431252dca314172ce2d0d945ff4d001af6a18f Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 21 二月 2024 19:39:16 +0800
Subject: [PATCH] update trainer
---
funasr/train_utils/trainer.py | 4 ++--
/dev/null | 12 ------------
examples/industrial_data_pretraining/seaco_paraformer/demo.sh | 0
examples/industrial_data_pretraining/fsmn_vad_streaming/demo.sh | 0
examples/industrial_data_pretraining/paraformer_streaming/demo.sh | 0
examples/industrial_data_pretraining/scama/demo.sh | 0
examples/industrial_data_pretraining/uniasr/demo.sh | 0
examples/industrial_data_pretraining/bicif_paraformer/demo.sh | 0
examples/industrial_data_pretraining/monotonic_aligner/demo.sh | 0
examples/industrial_data_pretraining/ct_transformer/demo.sh | 0
examples/industrial_data_pretraining/conformer/demo.sh | 0
examples/industrial_data_pretraining/contextual_paraformer/demo.sh | 0
examples/industrial_data_pretraining/ct_transformer_streaming/demo.sh | 0
examples/industrial_data_pretraining/paraformer-zh-spk/demo.sh | 0
14 files changed, 2 insertions(+), 14 deletions(-)
diff --git a/examples/industrial_data_pretraining/bicif_paraformer/infer.sh b/examples/industrial_data_pretraining/bicif_paraformer/demo.sh
similarity index 100%
rename from examples/industrial_data_pretraining/bicif_paraformer/infer.sh
rename to examples/industrial_data_pretraining/bicif_paraformer/demo.sh
diff --git a/examples/industrial_data_pretraining/conformer/infer.sh b/examples/industrial_data_pretraining/conformer/demo.sh
similarity index 100%
rename from examples/industrial_data_pretraining/conformer/infer.sh
rename to examples/industrial_data_pretraining/conformer/demo.sh
diff --git a/examples/industrial_data_pretraining/contextual_paraformer/infer.sh b/examples/industrial_data_pretraining/contextual_paraformer/demo.sh
similarity index 100%
rename from examples/industrial_data_pretraining/contextual_paraformer/infer.sh
rename to examples/industrial_data_pretraining/contextual_paraformer/demo.sh
diff --git a/examples/industrial_data_pretraining/ct_transformer/infer.sh b/examples/industrial_data_pretraining/ct_transformer/demo.sh
similarity index 100%
rename from examples/industrial_data_pretraining/ct_transformer/infer.sh
rename to examples/industrial_data_pretraining/ct_transformer/demo.sh
diff --git a/examples/industrial_data_pretraining/ct_transformer_streaming/infer.sh b/examples/industrial_data_pretraining/ct_transformer_streaming/demo.sh
similarity index 100%
rename from examples/industrial_data_pretraining/ct_transformer_streaming/infer.sh
rename to examples/industrial_data_pretraining/ct_transformer_streaming/demo.sh
diff --git a/examples/industrial_data_pretraining/emotion2vec/infer.sh b/examples/industrial_data_pretraining/emotion2vec/infer.sh
deleted file mode 100644
index df990b9..0000000
--- a/examples/industrial_data_pretraining/emotion2vec/infer.sh
+++ /dev/null
@@ -1,12 +0,0 @@
-
-#model="damo/emotion2vec_base"
-model="iic/emotion2vec_base_finetuned"
-model_revision="v2.0.4"
-
-python funasr/bin/inference.py \
-+model=${model} \
-+model_revision=${model_revision} \
-+input="https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav" \
-+output_dir="./outputs/debug" \
-+extract_embedding=False \
-+device="cpu" \
diff --git a/examples/industrial_data_pretraining/fsmn_vad_streaming/infer.sh b/examples/industrial_data_pretraining/fsmn_vad_streaming/demo.sh
similarity index 100%
rename from examples/industrial_data_pretraining/fsmn_vad_streaming/infer.sh
rename to examples/industrial_data_pretraining/fsmn_vad_streaming/demo.sh
diff --git a/examples/industrial_data_pretraining/monotonic_aligner/infer.sh b/examples/industrial_data_pretraining/monotonic_aligner/demo.sh
similarity index 100%
rename from examples/industrial_data_pretraining/monotonic_aligner/infer.sh
rename to examples/industrial_data_pretraining/monotonic_aligner/demo.sh
diff --git a/examples/industrial_data_pretraining/paraformer-zh-spk/infer.sh b/examples/industrial_data_pretraining/paraformer-zh-spk/demo.sh
similarity index 100%
rename from examples/industrial_data_pretraining/paraformer-zh-spk/infer.sh
rename to examples/industrial_data_pretraining/paraformer-zh-spk/demo.sh
diff --git a/examples/industrial_data_pretraining/paraformer_streaming/infer.sh b/examples/industrial_data_pretraining/paraformer_streaming/demo.sh
similarity index 100%
rename from examples/industrial_data_pretraining/paraformer_streaming/infer.sh
rename to examples/industrial_data_pretraining/paraformer_streaming/demo.sh
diff --git a/examples/industrial_data_pretraining/scama/infer.sh b/examples/industrial_data_pretraining/scama/demo.sh
similarity index 100%
rename from examples/industrial_data_pretraining/scama/infer.sh
rename to examples/industrial_data_pretraining/scama/demo.sh
diff --git a/examples/industrial_data_pretraining/seaco_paraformer/infer.sh b/examples/industrial_data_pretraining/seaco_paraformer/demo.sh
similarity index 100%
rename from examples/industrial_data_pretraining/seaco_paraformer/infer.sh
rename to examples/industrial_data_pretraining/seaco_paraformer/demo.sh
diff --git a/examples/industrial_data_pretraining/uniasr/infer.sh b/examples/industrial_data_pretraining/uniasr/demo.sh
similarity index 100%
rename from examples/industrial_data_pretraining/uniasr/infer.sh
rename to examples/industrial_data_pretraining/uniasr/demo.sh
diff --git a/funasr/train_utils/trainer.py b/funasr/train_utils/trainer.py
index 6a59f91..d175fbe 100644
--- a/funasr/train_utils/trainer.py
+++ b/funasr/train_utils/trainer.py
@@ -181,7 +181,7 @@
time2 = time.perf_counter()
time_escaped = (time2 - time1)/3600.0
- print(f"\nrank: {self.local_rank}, time_escaped_epoch: {time_escaped:.3f} hours, estimated to finish {self.max_epoch} epoch: {(self.max_epoch-epoch)*time_escaped:.3f}\n")
+ print(f"\nrank: {self.local_rank}, time_escaped_epoch: {time_escaped:.3f} hours, estimated to finish {self.max_epoch} epoch: {(self.max_epoch-epoch)*time_escaped:.3f} hours\n")
if self.rank == 0:
average_checkpoints(self.output_dir, self.avg_nbest_model)
@@ -293,7 +293,7 @@
f"{time_now}, "
f"rank: {self.local_rank}, "
f"epoch: {epoch}/{self.max_epoch}, "
- f"step: {batch_idx+1}/{len(self.dataloader_train)}, total: {self.batch_total}, "
+ f"step: {batch_idx+1}/{len(self.dataloader_train)}, total step: {self.batch_total}, "
f"(loss: {loss.detach().cpu().item():.3f}), "
f"(lr: {lr:.3e}), "
f"{[(k, round(v.cpu().item(), 3)) for k, v in stats.items()]}, "
--
Gitblit v1.9.1