Не удается загрузить модель Tensorflow v1 в tenorflow V2 с кодом обновления?

Question

Я сохранил модель в формате .ckpt и .h5, где моя папка содержит 4 файла.

, если я загружаю в версии 1, используя команду

model.load("model.ckpt") or model.load("model.h5")

, и загрузка успешно выполняется в Tensorflow Version1, но при загрузке в тензор потока версии 2 иногда отображается ошибка

  tf.keras.models.load_model('model.h5') or  tf.keras.models.load_model('model.ckpt')

выдача ошибки

SavedModel file does not exist at: /model/model.ckpt/{saved_model.pbtxt|saved_model.pb}

или иногда выдача ошибки

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

я дважды проверяю путь, загружаю в google colab и указываю точное местоположение, но не работаю.

вот мой кодовая ссылка

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

Answer 1

Если файл находится в папке на вашем диске, добавьте имя вашей папки в путь. в противном случае используйте код как есть.

import os
from google.colab import drive
drive.mount('/content/gdrive')
model_file_name = os.path.join("/content/gdrive/My Drive", 'model.h5')# CHANGE MODEL NAME
model = tf.keras.models.load_model(model_file_name)

При выполнении он покажет ссылку, щелкните по ней и выберите свою учетную запись на диске, где находится файл, а затем скопируйте и вставьте ключ в свою программу.