如何加载和继续在Google云端硬盘中保存为.H5的培训模型

时间:2019-07-09 04:19:42

标签: python tensorflow google-colaboratory pydrive

我按照Google Colab中的以下教程创建了一个生成RNN的文本:https://www.tensorflow.org/tutorials/sequences/text_generation 然后,我用自己的数据对其进行了训练。最后,我还添加了以下代码,以将其作为.h5文件保存到我的Google云端硬盘中,并在驱动器中创建了一个文件。

model.save('my_model.h5')
uploaded = drive.CreateFile({'title': 'FILE NAME HERE.h5'})
uploaded.SetContentFile('my_model.h5')
uploaded.Upload()
print('Uploaded file with ID {}'.format(uploaded.get('id')))

然后,我打开一个新笔记本并尝试像这样加载它:

downloaded = drive.CreateFile({'id': "FILE ID HERE"})
downloaded.GetContentFile('my_model.h5')
new_model = keras.models.load_model(downloaded)
new_model.summary()

但是,它给了我错误,其中包括:“'utf-8'编解码器无法解码位置0的字节0x89:无效的起始字节”

我已经尝试了其他文章,这些文章演示了如何实现我的目标,这就是我所得到的。

我希望能够继续训练模型,如果可能的话,不用原始代码。我该怎么办?

0 个答案:

没有答案