Предварительный тренинг модели Keras

Question

При запуске этой строки:

base_model = MobileNetV2(include_top=False, weights='imagenet', input_tensor=input_tensor, pooling=None, classes=NB_CLASS)

Я получаю следующую ошибку

Как мне решить проблему?

Answer 1

Удалось решить проблему путем удаления предварительно обученных весов и повторного запуска кода для повторной загрузки файла.

Файл предварительно обученных весов был сохранен в / home в скрытой папке /.keras в файле .h5 .

Answer 2

Без сохранения и загрузки файла в ваш код трудно сказать, в чем проблема. Если я хорошо понял, вы сохранили свои веса в файле h5py?

Я думаю, что всегда легче сохранить целую обученную модель, чем только веса:

from keras.models import load_model
model.save('my_model.h5')  # creates a HDF5 file 'my_model.h5'

# returns a compiled model
# identical to the previous one
model = load_model('my_model.h5')