将本地训练的 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?