From 31db90afbcc7a4feb6ed1f93c997d1134c2c2a7a Mon Sep 17 00:00:00 2001 From: anzhengqi <anzhengqi1@huawei.com> Date: Fri, 26 Aug 2022 15:13:05 +0800 Subject: [PATCH] modify some network readme and scripts --- official/cv/vit/README.md | 2 +- research/audio/tacotron2/src/tacotron2.py | 9 +++------ research/cv/ResNeSt50/README.md | 8 ++++---- research/cv/tsm/README_CN.md | 13 +++++-------- research/nlp/ktnet/scripts/run_record_eval.sh | 7 ++++++- 5 files changed, 19 insertions(+), 20 deletions(-) diff --git a/official/cv/vit/README.md b/official/cv/vit/README.md index 1024714e6..05244dfe2 100644 --- a/official/cv/vit/README.md +++ b/official/cv/vit/README.md @@ -410,7 +410,7 @@ Current batch_ Size can only be set to 1. | Parameters | Ascend | | -------------------------- | ----------------------------------------------------------- | | Model Version | Vit | -| Resource | Ascend 910; CPU 2.60GHz, 56cores; Memory 314G; OS Euler2.8 | +| Resource | Ascend 910; CPU 2.60GHz, 192cores; Memory 755G; OS Euler2.8 | | uploaded Date | 08/30/2021 (month/day/year) | | MindSpore Version | 1.3.0 | | Dataset | 1200k images | diff --git a/research/audio/tacotron2/src/tacotron2.py b/research/audio/tacotron2/src/tacotron2.py index 71f7922f2..9ca043b83 100644 --- a/research/audio/tacotron2/src/tacotron2.py +++ b/research/audio/tacotron2/src/tacotron2.py @@ -1223,10 +1223,7 @@ class TrainStepWrap(nn.Cell): overflow = self.loss_scaling_manager(self.loss_scale, cond) - if overflow: - succ = False - else: - succ = self.optimizer(grads) + if not overflow: + self.optimizer(grads) - ret = (loss, scale_sense) - return F.depend(ret, succ) + return loss, scale_sense diff --git a/research/cv/ResNeSt50/README.md b/research/cv/ResNeSt50/README.md index c9c025e95..6ff843b19 100644 --- a/research/cv/ResNeSt50/README.md +++ b/research/cv/ResNeSt50/README.md @@ -158,9 +158,9 @@ GPU: ```shell Ascend: # 鍒嗗竷寮忚缁冪ず渚嬶紙8鍗★級 - bash scripts/run_distribute_train.sh RANK_TABLE_FILE OUTPUT_DIR + bash run_distribute_train.sh RANK_TABLE_FILE # 鍗曟満璁粌 - bash scripts/run_train.sh OUTPUT_DIR + bash run_train.sh OUTPUT_DIR GPU: # 鍒嗗竷寮忚缁冪ず渚嬶紙8鍗★級 bash scripts/run_distribute_train_gpu.sh [DEVICE_NUM] @@ -172,9 +172,9 @@ GPU: ```shell # Ascend鍒嗗竷寮忚缁冪ず渚嬶紙8鍗★級 -bash scripts/run_distribute_train.sh RANK_TABLE_FILE OUTPUT_DIR +bash run_distribute_train.sh RANK_TABLE_FILE # Ascend鍗曟満璁粌绀轰緥 -bash scripts/run_train.sh OUTPUT_DIR +bash run_train.sh OUTPUT_DIR # GPU鍒嗗竷寮忚缁冪ず渚嬶紙8鍗★級 bash scripts/run_distribute_train_gpu.sh 8 # GPU鍗曟満璁粌绀轰緥 diff --git a/research/cv/tsm/README_CN.md b/research/cv/tsm/README_CN.md index b2f00eaa9..f262948be 100644 --- a/research/cv/tsm/README_CN.md +++ b/research/cv/tsm/README_CN.md @@ -46,14 +46,11 @@ TSM搴旂敤浜嗕竴绉嶉€氱敤鑰屾湁鏁堢殑鏃堕棿杞Щ妯″潡銆� 鏃堕棿杞Щ妯″潡灏� # 鏁版嵁闆� -浣跨敤鐨勬暟鎹泦锛歔SomethingSometing-v2](https://developer.qualcomm.com/software/ai-datasets/something-something) - -- 鏁版嵁闆嗗ぇ灏忥細19.5G锛屽叡174涓被銆�220,847娈佃棰� - - 璁粌闆嗭細16G锛屽叡168,913娈佃棰� - - 娴嬭瘯闆嗭細3.5G锛屽叡24,777娈佃棰� -- 鏁版嵁鏍煎紡锛歐ebm鏍煎紡瑙嗛 - - 娉細鏁版嵁灏嗗湪src/tools/gen_label_sthv2.py涓敓鎴愮被鏍囩銆� - - 娉細鏁版嵁灏嗗湪src/tools/vid2img_sthv2.py涓В鏋愭垚jpg鏍煎紡鍥惧儚銆� +浣跨敤鐨勬暟鎹泦锛歔SomethingSometing-v2](https://download.mindspore.cn/dataset/somethingv2.tar) + +- 鏁版嵁闆嗗ぇ灏忥細281G锛屽叡174涓被銆�220,847娈佃棰� + - 璁粌闆嗭細鍏�168,913娈佃棰� + - 娴嬭瘯闆嗭細鍏�24,777娈佃棰� # 鐗规€� diff --git a/research/nlp/ktnet/scripts/run_record_eval.sh b/research/nlp/ktnet/scripts/run_record_eval.sh index 02bab108d..06e762a5c 100644 --- a/research/nlp/ktnet/scripts/run_record_eval.sh +++ b/research/nlp/ktnet/scripts/run_record_eval.sh @@ -15,6 +15,7 @@ # ============================================================================ PWD_DIR=`pwd` +scripts_path=$(dirname $0) DATA=$1 LOAD_CHECKPOINT_PATH=$2 @@ -22,7 +23,11 @@ BERT_DIR=$DATA/cased_L-24_H-1024_A-16 WN_CPT_EMBEDDING_PATH=$DATA/KB_embeddings/wn_concept2vec.txt NELL_CPT_EMBEDDING_PATH=$DATA/KB_embeddings/nell_concept2vec.txt -python3 run_KTNET_record_eval.py \ +if [ ! -d log ]; then + mkdir log +fi + +python3 $scripts_path/../run_KTNET_record_eval.py \ --device_target "Ascend" \ --device_id 0 \ --batch_size 12 \ -- GitLab