From 89feb45e176ff777da24a9e18f0e6eaeb20bc5e8 Mon Sep 17 00:00:00 2001
From: anzhengqi <anzhengqi1@huawei.com>
Date: Fri, 17 Jun 2022 17:48:29 +0800
Subject: [PATCH] modify networks

---
 research/cv/SE-Net/export.py                            | 2 +-
 research/cv/u2net/export.py                             | 1 -
 research/cv/vit_base/src/modeling_ms.py                 | 5 +----
 research/cv/yolov3_tiny/scripts/run_standalone_train.sh | 4 +++-
 4 files changed, 5 insertions(+), 7 deletions(-)

diff --git a/research/cv/SE-Net/export.py b/research/cv/SE-Net/export.py
index 5453b9ed4..57ffe5aa8 100644
--- a/research/cv/SE-Net/export.py
+++ b/research/cv/SE-Net/export.py
@@ -28,7 +28,7 @@ if config.device_target == "Ascend":
 def run_export():
     """run export."""
     if config.network_dataset == 'se-resnet50_imagenet2012':
-        from src.resnet import resnet50 as resnet
+        from src.resnet import se_resnet50 as resnet
     elif config.network_dataset == 'se-resnet101_imagenet2012':
         from src.resnet import resnet101 as resnet
     else:
diff --git a/research/cv/u2net/export.py b/research/cv/u2net/export.py
index 1b3bb1a1c..e48d13b75 100644
--- a/research/cv/u2net/export.py
+++ b/research/cv/u2net/export.py
@@ -31,7 +31,6 @@ args = parser.parse_args()
 context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target)
 
 if __name__ == '__main__':
-    context.set_context(device_id="Ascend")
     net = U2NET()
     param_dict = load_checkpoint(args.ckpt_file)
     load_param_into_net(net, param_dict)
diff --git a/research/cv/vit_base/src/modeling_ms.py b/research/cv/vit_base/src/modeling_ms.py
index dba23f9fc..a81a5b434 100644
--- a/research/cv/vit_base/src/modeling_ms.py
+++ b/research/cv/vit_base/src/modeling_ms.py
@@ -27,9 +27,6 @@ def swish(x):
     return x * P.Sigmoid()(x)
 
 
-ACT2FN = {"gelu": nn.GELU(), "relu": P.ReLU(), "swish": swish}
-
-
 class Attention(nn.Cell):
     """Attention"""
     def __init__(self, config):
@@ -87,7 +84,7 @@ class Mlp(nn.Cell):
                             weight_init='XavierUniform', bias_init='Normal')
         self.fc2 = nn.Dense(config.transformer_mlp_dim, config.hidden_size,
                             weight_init='XavierUniform', bias_init='Normal')
-        self.act_fn = ACT2FN["gelu"]
+        self.act_fn = nn.GELU()
         self.dropout = nn.Dropout(config.transformer_dropout_rate)
 
     def construct(self, x):
diff --git a/research/cv/yolov3_tiny/scripts/run_standalone_train.sh b/research/cv/yolov3_tiny/scripts/run_standalone_train.sh
index 7285ac714..29adf4b1e 100644
--- a/research/cv/yolov3_tiny/scripts/run_standalone_train.sh
+++ b/research/cv/yolov3_tiny/scripts/run_standalone_train.sh
@@ -50,7 +50,9 @@ then
 fi
 mkdir ./train
 cp ../*.py ./train
+cp ../*.yaml ./train
 cp -r ../src ./train
+cp -r ../model_utils ./train
 cd ./train || exit
 echo "start training for device $DEVICE_ID"
 env > env.log
@@ -66,4 +68,4 @@ python train.py \
     --per_batch_size=32 \
     --weight_decay=0.016 \
     --lr_scheduler=cosine_annealing > log.txt 2>&1 &
-cd ..
\ No newline at end of file
+cd ..
-- 
GitLab