如何直接從雲存儲桶加載 tf.keras model？

Question

我嘗試直接從雲存儲桶加載 tf.keras model 但我看不到這樣做很容易。 我想加載整個 model 結構，而不僅僅是權重。

我看到了 3 個可能的方向：

1. 是否可以直接從谷歌雲存儲桶加載 keras model ？ 命令 tf.keras.model.load_model('gs://my_bucket/model.h5') 不起作用

2. 我嘗試使用 tensorflow.python.lib.ii.file_io 但我不知道如何將其加載為 model。

3. 我通過 gsutil cp 命令將 model 復制到本地目錄，但我不知道如何等到操作完成。 tf 嘗試在下載操作完成之前加載 model 因此出現錯誤

我會感謝任何建議。

彼得

Answer 1

1. 從 gs 存儲加載文件

from tensorflow.python.lib.io import file_io
model_file = file_io.FileIO('gs://mybucket/model.h5', mode='rb')

2. 在本地保存 model 的臨時副本

temp_model_location = './temp_model.h5'
temp_model_file = open(temp_model_location, 'wb')
temp_model_file.write(model_file.read())
temp_model_file.close()
model_file.close()

3. 加載本地保存的model

model = tf.keras.models.load_model(temp_model_location)