Skip to content
Snippets Groups Projects
Commit 31db90af authored by anzhengqi's avatar anzhengqi
Browse files

modify some network readme and scripts

parent c2f8156c
No related branches found
No related tags found
No related merge requests found
......@@ -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 |
......
......@@ -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
......@@ -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单机训练示例
......
......@@ -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段视频
- 数据格式:Webm格式视频
- 注:数据将在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段视频
# 特性
......
......@@ -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 \
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment