2016-12-16 119 views
1

我正在通过来自here的泰坦尼克号教程。完成后,我想保存模型,然后再加载它。保存之后,我得到一个警告提示:无法使用TensorFlow和TFLEARN保存或加载模型

TensorFlow's V1 checkpoint format has been deprecated. 
WARNING:tensorflow:Consider switching to the more efficient V2 format: 

当我打开它,我得到一个错误说

NotFoundError (see above for traceback): Unsuccessful TensorSliceReader constructor: Failed to find any matching files for model.tflearn 
[[Node: save_1/RestoreV2_2 = RestoreV2[dtypes=[DT_FLOAT], _device="/job:localhost/replica:0/task:0/cpu:0"](_recv_save_1/Const_0, save_1/RestoreV2_2/tensor_names, save_1/RestoreV2_2/shape_and_slices)]] 

这是我如何保存/载入模型。

net = tflearn.input_data(shape=[None, 6]) 
net = tflearn.fully_connected(net, 32) 
net = tflearn.fully_connected(net, 32) 
net = tflearn.fully_connected(net, 2, activation='softmax') 
net = tflearn.regression(net) 


model = tflearn.DNN(net) 

model.fit(data, labels, n_epoch=10, batch_size=16, show_metric=True) 
model.save('model.tflearn') 

,然后当我打开它,我只是做了...

model.load('model.tflearn') 

如果任何人谁拥有经验,这可能给我的手,我将不胜感激,谢谢。

回答

1

使用model.load('./model.tflearn')

+1

请不要忘记解释/评论你的代码,并提供相关文件[来自评论] – Blag

相关问题