Question

我正在阅读here的泰坦尼克号教程。完成后我想保存模型，然后再加载它。保存后我收到警告说：

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

如果有任何有此经验的人可以帮助我，我会很感激，谢谢。

Answer 1

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

使用TensorFlow＆amp ;;无法保存或加载模型TFLEARN

1 个答案: