diff --git a/examples/model_security/model_attacks/cv/faster_rcnn/src/FasterRcnn/fpn_neck.py b/examples/model_security/model_attacks/cv/faster_rcnn/src/FasterRcnn/fpn_neck.py index 73781bd..6995adf 100644 --- a/examples/model_security/model_attacks/cv/faster_rcnn/src/FasterRcnn/fpn_neck.py +++ b/examples/model_security/model_attacks/cv/faster_rcnn/src/FasterRcnn/fpn_neck.py @@ -89,7 +89,7 @@ class FeatPyramidNeck(nn.Cell): self.interpolate1 = P.ResizeNearestNeighbor((48, 80)) self.interpolate2 = P.ResizeNearestNeighbor((96, 160)) self.interpolate3 = P.ResizeNearestNeighbor((192, 320)) - self.maxpool = P.MaxPool(ksize=1, strides=2, padding="same") + self.maxpool = P.MaxPool(kernel_size=1, strides=2, pad_mode="same") def construct(self, inputs): x = () diff --git a/examples/model_security/model_attacks/cv/faster_rcnn/src/FasterRcnn/resnet50.py b/examples/model_security/model_attacks/cv/faster_rcnn/src/FasterRcnn/resnet50.py index eb0fd57..20a2168 100644 --- a/examples/model_security/model_attacks/cv/faster_rcnn/src/FasterRcnn/resnet50.py +++ b/examples/model_security/model_attacks/cv/faster_rcnn/src/FasterRcnn/resnet50.py @@ -89,7 +89,7 @@ class ResNetFea(nn.Cell): self.conv1 = _conv(3, 64, kernel_size=7, stride=2, padding=3, pad_mode='pad') self.bn1 = _BatchNorm2dInit(64, affine=bn_training, use_batch_statistics=bn_training) self.relu = P.ReLU() - self.maxpool = P.MaxPool(ksize=3, strides=2, padding="SAME") + self.maxpool = P.MaxPool(kernel_size=3, strides=2, pad_mode="SAME") self.weights_update = weights_update if not self.weights_update: