Я тренировал свой CNN (VGG) через google colab и сгенерировал файл.h5. Теперь проблема в том, что я могу предсказать свой результат успешно через google colab, но когда я загружаю этот файл.h5, подготовленный файл модели и пытаюсь предсказать вывод на моем ноутбуке, я получаю ошибку при загрузке модели.
Вот код:
import tensorflow as tf
from tensorflow import keras
import h5py
# Initialization
loaded_model = keras.models.load_model('./train_personCount_model.h5')
И ошибка:
ValueError: Unknown initializer: GlorotUniform