使用tflearn

时间:2018-05-25 14:40:53

标签: python tensorflow keras tflearn

我在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生成的文件,但它不起作用。

0 个答案:

没有答案