2016-12-16 3 views
1

Je suis en train de lire le tutoriel Titanic de here. Après l'avoir terminé, je voulais sauvegarder le modèle et le charger plus tard. Après avoir enregistré, je reçois un avertissement disant:Impossible d'enregistrer ou de charger des modèles avec TensorFlow & TFLEARN

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

Quand je charge, je reçois une erreur disant

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)]] 

Voilà comment je CONSERVE/chargé le modèle.

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') 

puis quand je le charge je fais juste un ...

model.load('model.tflearn') 

Si quelqu'un qui a une expérience avec cela pourrait me donner un coup de main je l'apprécierais, merci.

Répondre

1

utilisation model.load('./model.tflearn')

+1

S'il vous plaît ne pas oublier d'expliquer/commenter votre code, et de fournir la documentation pertinente [de l'examen] – Blag