加载Keras模型时出现未知的初始化器:GlorotUniform。

6 浏览
0 Comments

加载Keras模型时出现未知的初始化器:GlorotUniform。

我通过谷歌协作平台训练了我的卷积神经网络(VGG),并生成了.h5文件。现在的问题是,当我在谷歌协作平台上成功预测输出时,但是当我下载该.h5训练模型文件并尝试在我的笔记本电脑上进行预测输出时,加载模型时出现错误。以下是代码:

import tensorflow as tf
from tensorflow import keras
import h5py
# 初始化
loaded_model = keras.models.load_model('./train_personCount_model.h5')

错误信息如下:

ValueError: 未知的初始化器:GlorotUniform

0