我正在尝试在aws lambda上使用keras vgg16模型进行预测。但是我得到了一个错误:尝试下载模型时没有空间了。
from keras.applications.imagenet_utils import preprocess_input
from keras.applications import VGG16, nasnet, xception, inception_v3
from keras.preprocessing import image
from keras.applications.imagenet_utils import preprocess_input
from keras.models import Model
import json
def loadmodel(use_model):
m_model = VGG16(weights = 'imagenet')
last_layer = Model(inputs=m_model.input,
outputs=m_model.get_layer("fc2").output)
return m_model_sans_last_layer
model_load=loadmodel("m_VGG16")
我得到了错误:
[Errno 28] No space left on device
我正在使用3000 mb lambda内存。
如何在AWS Lambda上使用该模型?
我可以保存最后一层并仅加载最后一层进行预测吗?