我在Keras训练过模型,我需要在Tflearn制作的代码中加载这些模型。
我可以使用以下行保存Keras训练的模型:
model.save('my_model.hdf5') #keras
它保存了训练模型的架构和权重。
或行:
model.save_weights('my_model.hdf5') #keras
它仅保存训练模型的权重。
目的是使用例如以下行将模型加载到tflearn中:
model.load ('example.tflearn') #tflearn
这一行只是加载权重,要求之前传递架构。
我尝试使用.load of tflearn从Keras加载.save和.save_weights生成的文件,但它不起作用。