【发布时间】:2019-08-10 22:41:59
【问题描述】:
我使用 GPU 从 Linux 训练了一个模型,并使用 model.save('my_model.h5') 保存它。
使用
在 Linux 机器中打开 .h5 模型没有问题from keras.models import load_model
model = load_model('my_model.h5')
但是,当我在我的 Mac 上加载它时遇到了问题(它没有 GPU,而且可能 tensorflow 也不是 GPU 版本的那个):
OSError: Unable to open file (unable to open file: name = 'model0.h5', errno = 2, error message = 'No such file or directory', flags = 0, o_flags = 0)
当我在 jupyter notebook 上工作时,相同的命令,相同的位置,存储 .h5 文件的相同位置,我想要加载的相同文件。有谁知道发生了什么?
提前谢谢你。
【问题讨论】:
-
试试
os.path.exists('my_model.h5')或者os.listdir('.')或os.getcwd()
标签: python macos keras save load