2016-12-16 10 views
1

Я изучаю учебник Titanic от 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