I'm following a tutorial and loaded modelo VGGNet16
pre-trained using Keras
vgg16_model = keras.applications.vgg16.VGG16()
model = Sequential()
for layer in vgg16_model.layers:
model.add(layer)
model.layers.pop()
for layer in model.layers:
layer.trainable = False
model.add(Dense(10, activation='softmax', name='predict'))
#model.summary()
I used model.save('path/model_1.h5')
to save the template after the train with model.fit_generator(...)
So I ran out of time on "Colaboratory". so I wanted to use model = load_model('path/model_1.h5')
to reload my template rather than load showed as previously with vgg16_model = keras.applications.vgg16.VGG16()...
And now I have this error:
ValueError: Dimension 0 in both shapes must be equal, but are 4096 and 1000. Shapes are [4096,10] and [1000,10]. for 'Assign_61' (op: 'Assign') with input shapes: [4096,10], [1000,10].
What am I doing wrong? Thanks!