Last gelerntes Modell in anderen Verfahren mit tflearn

Question

Ich versuche, ein Modell zu speichern und in anderen Verfahren mit tflearn Bibliothek laden ...

So erzeugen ich das Modell:

lenx = 21908 
leny = 81 
# Build neural network 
net = tflearn.input_data(shape=[None, lenx]) 
net = tflearn.fully_connected(net, 8) 
net = tflearn.fully_connected(net, 8) 
net = tflearn.fully_connected(net, leny, activation='softmax') 
net = tflearn.regression(net) 

# Define model and setup tensorboard 
model = tflearn.DNN(net, tensorboard_dir='tflearn_logs') 
# Start training (apply gradient descent algorithm) 
model.fit(train_x, train_y, n_epoch=10, batch_size=8, show_metric=True) 
model.save('model.tflearn') 

, das funktioniert OK! dann in anderer Datei zu laufen, in anderem Prozess, den ich es zu laden bin versucht, auf diese Weise:

lenx = 21908 
leny = 81 
# Build neural network 
net = tflearn.input_data(shape=[None, lenx]) 
net = tflearn.fully_connected(net, 8) 
net = tflearn.fully_connected(net, 8) 
net = tflearn.fully_connected(net, leny, activation='softmax') 
net = tflearn.regression(net) 

model = tflearn.DNN(net, tensorboard_dir='tflearn_logs') 

model.load("model.tflearn") 

Aber ich habe diesen Fehler:

ValueError: Cannot feed value of shape (1, 0) for Tensor 'InputData/X:0', which has shape '(?, 21908)' 

ich viele Dinge ausprobiert, aber es tut nicht arbeiten.

Answer 1

denke ich, was Sie fehlt ein „weights_only = True“ ist Argument in der Last:

model.load("model.tflearn", weights_only=True)