将本地训练的 TensorFlow 模型导入 Google Colab

Posted

技术标签:

【中文标题】将本地训练的 TensorFlow 模型导入 Google Colab【英文标题】:Importing Locally Trained TensorFlow Model to Google Colab 【发布时间】:2022-01-08 13:47:47 【问题描述】:

我正在 Google Magenta (MusicVAE) 上训练我自己的模型,我正在使用自己的音乐文件并在我的计算机上本地训练模型。我无法弄清楚如何将此模型加载到 Google Colab 中以使用该模型。有谁知道如何将本地训练的模型加载到 Google Colab 上?

【问题讨论】:

【参考方案1】:

过去,我曾将我的文件上传到 Google 云端硬盘,然后将我的云端硬盘安装到 Colab 以访问这些文件。我为此寻找文档,并找到了这个页面,该页面还解释了如何直接上传文件: https://colab.research.google.com/notebooks/io.ipynb

其中一种解决方案对您有用吗?

一些额外说明:

如果模型很大,可能无法上传到云端硬盘 确保在 Colab 中使用相同的 Python/TensorFlow 版本

【讨论】:

以上是关于将本地训练的 TensorFlow 模型导入 Google Colab的主要内容,如果未能解决你的问题,请参考以下文章

如何使用 tf.estimator 导入保存的 Tensorflow 模型训练并预测输入数据

TensorFlow基础 —— 模型的保存读取与可视化方法总结

Tensorflow 训练模型在云端机器上工作,但在我的本地电脑上使用时出错

如何在不使用本地PC转换的情况下将我在google colab中训练的tensorflow的protobuf训练模型转换为tensorflowjs?

如何使用Tensorflow构建Seq2seq模型

tensorflow 保存好训练的模型,restore 后验证集的正确率很低