From fd022b931ab9d8bae6a9933121dca6a80d16c9d9 Mon Sep 17 00:00:00 2001 From: zhouneng2 <zhouneng2@huawei.com> Date: Tue, 15 Feb 2022 17:55:43 +0800 Subject: [PATCH] [FasterRCNN]fix import error --- official/cv/faster_rcnn/eval.py | 1 + official/cv/faster_rcnn/src/network_define.py | 6 +++--- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/official/cv/faster_rcnn/eval.py b/official/cv/faster_rcnn/eval.py index 0e9294dec..86da7c5a7 100644 --- a/official/cv/faster_rcnn/eval.py +++ b/official/cv/faster_rcnn/eval.py @@ -20,6 +20,7 @@ from collections import defaultdict import numpy as np from pycocotools.coco import COCO +import mindspore as ms from mindspore.common import set_seed, Parameter from src.dataset import data_to_mindrecord_byte_image, create_fasterrcnn_dataset, parse_json_annos_from_txt diff --git a/official/cv/faster_rcnn/src/network_define.py b/official/cv/faster_rcnn/src/network_define.py index c7edbf325..cf62aa6ba 100644 --- a/official/cv/faster_rcnn/src/network_define.py +++ b/official/cv/faster_rcnn/src/network_define.py @@ -15,10 +15,10 @@ """FasterRcnn training network wrapper.""" import time -import numpy as np +import mindspore.common.dtype as mstype import mindspore.ops as ops import mindspore.nn as nn -from mindspore import ParameterTuple +from mindspore import ParameterTuple, Tensor from mindspore.train.callback import Callback from mindspore.nn.wrap.grad_reducer import DistributedGradReducer @@ -134,7 +134,7 @@ class TrainOneStepCell(nn.Cell): self.optimizer = optimizer self.grad = ops.GradOperation(get_by_list=True, sens_param=True) - self.sens = ms.numpy.ones((1,) * sens).astype(np.float32) + self.sens = Tensor([sens,], mstype.float32) self.reduce_flag = reduce_flag if reduce_flag: self.grad_reducer = DistributedGradReducer(optimizer.parameters, mean, degree) -- GitLab