!13609 add kwargs for vgg16

From: @caojian05
Reviewed-by: @wuxuejian,@oacjiewen
Signed-off-by: @wuxuejian
pull/13609/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit baa35ab3f9

@ -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