在Tensorflow 2.x中加载Tensorflow 1.x模型时出现错误

时间:2020-01-23 06:48:01

标签: machine-learning nlp data-science tensorflow2.0

我正在将Tensorflow 1.x的保存模型加载到Tensorflow 2.x中,但是出现问题

我正在使用此代码

import os
import tensorflow as tf
from tensorflow import keras
from tensorflow.python.keras.layers import Dense,Flatten
from tensorflow.python.keras import Sequential
from google.colab import drive
drive.mount('/content/gdrive')
model_file_name = os.path.join("my_drive/AI_values/model", 'model.ckpt')# CHANGE MODEL NAME
model = tf.keras.models.load_model(model_file_name)

我遇到错误

OSError: Unable to open file (unable to open file: name = 'my_drive/AI_values/model/model.ckpt', 
errno = 2, error message = 'No such file or directory', flags = 0, o_flags = 0)

它显示的是操作系统错误,这是权限,也显示无法打开文件,但我同时尝试了.ckpt和.h5格式,但无法正常工作。

请帮我,我正在分享模型google驱动器链接

https://drive.google.com/open?id=1tPoUDM5Re3ZKXwK2f7kbzwK8Lad6vtCX

还有Google colab链接

https://colab.research.google.com/drive/1hf7AwEMJHf4zLZOBSzxejA2ISuTQ3N9K

1 个答案:

答案 0 :(得分:0)

您可以在

中将my_drive替换为My Drive(包括空格)吗?

model_file_name = os.path.join("my_drive/AI_values/model", 'model.ckpt')

尝试?