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