tensorflow-gpu 2.0.0中load_model()方法的报错和解决

调用Kears中load_model()方法遇到的问题和解决方法

运行环境：tensorflow-gpu 2.0.0a0，显卡配置：GTX1050Ti
报错：Unknown entry in loss dictionary: class_name. Only expected following keys:[‘dense_2’]

程序流程如下

执行以下代码进行网络训练，调用keras自带fit函数

model = build_model()

# Train
history = model.fit(img, label, batch_size=16,
                    epochs=5, validation_split=0.05)

其中，使用的部分网络参数如下

# configure the model
model.compile(optimizer=keras.optimizers.Adam(),
              loss=keras.losses.SparseCategoricalCrossentropy(),
              metrics=['accuracy'])

使用gpu训练，训练完成的模型可以在CPU版本的tensorflow 2.0.0通过load_model()正常载入和运行。
在gpu版本中调用以下函数

def load_pretrained_model(filename):
    pretrained_model = load_model(filename)
    return pretrained_model

出现报错：