我正在尝试创建一个神经网络并收集所有可能的信息,以进行比较。我想节省培训的每一步以及所花费的时间。每次我运行网络时,它都会打印一些带有此信息的信息消息,我想知道如何保存它。
我已经创建了自己的模型函数并设置了配置。
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)