From 4efa7ef4e144a457138bd2ba8078cdf0fa2a8d23 Mon Sep 17 00:00:00 2001 From: unknown Date: Thu, 27 May 2021 08:49:50 +0800 Subject: [PATCH] maskrcnn_mobilenetv1 export --- model_zoo/official/cv/maskrcnn_mobilenetv1/src/network_define.py | 1 + 1 file changed, 1 insertion(+) diff --git a/model_zoo/official/cv/maskrcnn_mobilenetv1/src/network_define.py b/model_zoo/official/cv/maskrcnn_mobilenetv1/src/network_define.py index c5adf060929..7825a19ebcc 100644 --- a/model_zoo/official/cv/maskrcnn_mobilenetv1/src/network_define.py +++ b/model_zoo/official/cv/maskrcnn_mobilenetv1/src/network_define.py @@ -23,6 +23,7 @@ from mindspore.ops import composite as C from mindspore import ParameterTuple from mindspore.train.callback import Callback from mindspore.nn.wrap.grad_reducer import DistributedGradReducer +from src.maskrcnn_mobilenetv1.mask_rcnn_mobilenetv1 import Mask_Rcnn_Mobilenetv1 time_stamp_init = False time_stamp_first = 0