我浏览了tensorflow文档,但找不到使用SavedModelBuilder类以最佳验证精度保存模型的方法。 我正在使用tflearn进行模型构建,下面是我尝试过的解决方法,但是这需要很多时间,在这里我分别为每个纪元运行fit方法并保存模型
for i in range(epoch):
model.fit(trainX, trainY, n_epoch=1, validation_set=(testX, testY), show_metric=True, batch_size=8)
builder = tf.saved_model.builder.SavedModelBuilder('/tmp/serving/model/' + str(i))
builder.add_meta_graph_and_variables(model.session,
['TRAINING'],
signature_def_map={
'predict': prediction_sig
})
builder.save()
请建议是否有更好的方法。在
明白了。它可以通过tflearn回调来实现。 谢谢。在
相关问题 更多 >
编程相关推荐