在colab训练的代码,换别的平台要怎么改

155次阅读

共计 596 个字符,预计需要花费 2 分钟才能阅读完成。

from google.colab import files 代码原版是这样 在别的平台训练显示 google colab 模块不存在

from google.colab import files 这行代码是 Colab 专用的, 其他平台上是不存在这个模块的。
要在其他平台上运行原先在 Colab 上训练的代码, 你需要:
删除 from google.colab import files 这一行
修改数据的读取方式: 不要使用 files.upload() 和 files.download(), 直接从本地路径读取数据
修改模型保存 / 加载的方式: 不要使用 files.download_blob() 和 files.upload_blob(), 使用本地文件路径
举个例子, 原来 Colab 上的代码:
from google.colab import files
data = files.upload()
Train model…
files.download_blob(model.to_json(), ‘model.json’)
修改后的代码:
data = np.load(‘data.npy’) # Load from local data.npy
Train model…
model.save(‘model.h5’) # Save to local model.h5
这样就可以在其他平台上运行 Colab 训练过的代码了。

正文完
 0