使用 Google Colab 训练模型后,我使用以下命令(在 Google Colab 中)下载了它:
model.save('model.h5')
from google.colab import files
files.download('model.h5')
Run Code Online (Sandbox Code Playgroud)
我的问题是,当我尝试使用本地机器(在 Google Colab 之外)加载下载的model.h5 时,出现以下错误:
[输入]
from keras.models import load_model
model = load_model(model.h5)
Run Code Online (Sandbox Code Playgroud)
[输出]
Traceback (most recent call last):
File "test.py", line 2, in <module>
model = load_model(filepath = 'saved_model/model2.h5',custom_objects=None,compile=True, )
File "/home/lucasmirachi/anaconda3/envs/myenviron/lib/python3.8/site-packages/tensorflow/python/keras/saving/save.py", line 184, in load_model
return hdf5_format.load_model_from_hdf5(filepath, custom_objects, compile)
File "/home/lucasmirachi/anaconda3/envs/myenviron/lib/python3.8/site-packages/tensorflow/python/keras/saving/hdf5_format.py", line 177, in load_model_from_hdf5
model = model_config_lib.model_from_config(model_config,
File "/home/lucasmirachi/anaconda3/envs/myenviron/lib/python3.8/site-packages/tensorflow/python/keras/saving/model_config.py", line 55, in model_from_config
return deserialize(config, custom_objects=custom_objects)
File "/home/lucasmirachi/anaconda3/envs/myenviron/lib/python3.8/site-packages/tensorflow/python/keras/layers/serialization.py", line 105, in deserialize …Run Code Online (Sandbox Code Playgroud)