如何在tf中保存培训过程的所有信息?

时间:2019-06-27 12:02:36

标签: python tensorflow

我正在尝试创建一个神经网络并收集所有可能的信息,以进行比较。我想节省培训的每一步以及所花费的时间。每次我运行网络时,它都会打印一些带有此信息的信息消息,我想知道如何保存它。

我已经创建了自己的模型函数并设置了配置。

estimator = tf.estimator.Estimator(model_fn=model_fn, config=config)

然后创建输入函数并对其进行了训练

input_fn = tf.estimator.inputs.numpy_input_fn(
            {"x": x_train[order,:]}, y_train[order],
            batch_size=30, num_epochs=None, shuffle=False)

estimator.train(input_fn=input_fn, max_steps=steps)

0 个答案:

没有答案