Python 如何将tensorflow模型保存到google drive
我正在google colab中使用tensor flow,我想将我的模型保存在google drive上以备将来使用,有人可以指导我如何操作吗?首先连接到google drive:Python 如何将tensorflow模型保存到google drive,python,tensorflow,google-colaboratory,Python,Tensorflow,Google Colaboratory,我正在google colab中使用tensor flow,我想将我的模型保存在google drive上以备将来使用,有人可以指导我如何操作吗?首先连接到google drive: from google.colab import drive drive.mount('/gdrive') 下一步将您的模型保存在colab上 model.save('modelname') 保存的模型现在将显示在“内容”文件夹中。现在,在colab的“gdrive”文件夹中创建相同的文件夹以及子文件夹。您可以
from google.colab import drive
drive.mount('/gdrive')
下一步将您的模型保存在colab上
model.save('modelname')
保存的模型现在将显示在“内容”文件夹中。现在,在colab的“gdrive”文件夹中创建相同的文件夹以及子文件夹。您可以使用shutil将相关文件复制到这些文件夹中
import shutil
shutil.copy('source','destination')
您可以使用“复制路径”选项来获取文件的确切地址。最后,要加载模型,必须连接到驱动器并加载模型
model = tf.keras.models.load_model('modeladdress')
只要在谷歌上搜索一下,你就会发现很多像@Rafaeldourado这样的页面。你提到的文章保存了一个Pytorch模型而不是Tensorflow,我在谷歌上找不到解决方案,这就是我问这个问题的原因。你的意思是
.tfite
或.h5