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