keras和pytorch对于引入网络模型有一些区别
pytorch
from torchvision import models
resnet_model = models.resnet18(pretrained=True)
# pretrained 设置为 True,会自动下载模型 所对应权重,并加载到模型中
# 也可以自己下载 权重,然后 load 到 模型中
keras
from keras.models import Model
from keras.applications.vgg16 import VGG16
from keras import layers
base_model = VGG16(weights =