Browse Source

!13609 add kwargs for vgg16

From: @caojian05
Reviewed-by: @wuxuejian,@oacjiewen
Signed-off-by: @wuxuejian
pull/13609/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
baa35ab3f9
2 changed files with 3 additions and 3 deletions
  1. +0
    -0
      mindspore/lite/micro/example/mnist_stm32f746/mnist/include/api/lite_context.h
  2. +3
    -3
      model_zoo/official/cv/vgg16/src/vgg.py

+ 0
- 0
mindspore/lite/micro/example/mnist_stm32f746/mnist/include/api/lite_context.h View File


+ 3
- 3
model_zoo/official/cv/vgg16/src/vgg.py View File

@@ -127,7 +127,7 @@ cfg = {
}


def vgg16(num_classes=1000, args=None, phase="train"):
def vgg16(num_classes=1000, args=None, phase="train", **kwargs):
"""
Get Vgg16 neural network with batch normalization.

@@ -140,11 +140,11 @@ def vgg16(num_classes=1000, args=None, phase="train"):
Cell, cell instance of Vgg16 neural network with batch normalization.

Examples:
>>> vgg16(num_classes=1000, args=args)
>>> vgg16(num_classes=1000, args=args, **kwargs)
"""

if args is None:
from .config import cifar_cfg
args = cifar_cfg
net = Vgg(cfg['16'], num_classes=num_classes, args=args, batch_norm=args.batch_norm, phase=phase)
net = Vgg(cfg['16'], num_classes=num_classes, args=args, batch_norm=args.batch_norm, phase=phase, **kwargs)
return net

Loading…
Cancel
Save