未知的初始值设定项:加载Keras模型时，GlorotUniform [英] Unknown initializer: GlorotUniform when loading Keras model

问题描述

我通过Google colab训练了CNN(VGG)，并生成了.h5文件.现在的问题是，我可以通过google colab成功预测我的输出，但是当我下载经过.h5训练的模型文件并尝试在笔记本电脑上预测输出时，加载模型时出现错误.

I trained my CNN (VGG) through google colab and generated .h5 file. Now problem is, I can predict my output successfully through google colab but when i download that .h5 trained model file and try to predict output on my laptop, I am getting error when loading the model.

这是代码:

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

推荐答案

我遇到了同样的问题.更改后:

I ran into the same issue. After changing:

from tensorflow import keras

收件人:

import keras

生活再次值得生活.

这篇关于未知的初始值设定项:加载Keras模型时，GlorotUniform的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！