Я использую модель VGG16 для создания классификатора на кошек и собак. Я изменил последний слой модели, чтобы иметь 2 узла вместо 1000 узлов. Я сохраняю модель здесь:
model.save ( 'CNNFromScratch.h5')
и загрузка его в другой файл:
new_model = load_model ('CNNFromScratch.h5')
И я получаю следующую ошибку:
Traceback (последний вызов был последним): File
"/Usr/local/lib/python3.5/site-packages/tensorflow/python/framework/ops.py",
строка 1567, в _create_c_op
c_op = c_api.TF_FinishOperation (op_desc) tenorflow.python.framework.errors_impl.InvalidArgumentError:
Размер 0 в обеих формах должен быть одинаковым, но равен 4096 и 1000.
Формы [4096,2] и [1000,2]. для 'Assign_30' (оп: 'Assign') с
входные формы: [4096,2], [1000,2].
.
Что я могу сделать, чтобы решить проблему?