mindspore_hub 可以使用gpu加载vgg16模型以及是否可以做迁移模型吗？

问题：

gpu加载vgg16模型过程中，希望不保留全连接层，但是vgg16 模型好像不包括include_top这个选项，请问有别的办法吗？谢谢大家

 

解决方案：

请手动修改规避，修改如下两点即可：

1.修改vgg.py 130行，增加**kwargs参数：如下

def vgg16(num_classes=1000, args=None, phase="train", **kwargs):

2.修改倒数第二行：增加**kwargs参数：如下

net = Vgg(cfg['16'], num_classes=num_classes, args=args, batch_norm=args.batch_norm, phase=phase, **kwargs)