如何在“适合”Tensorflow中禁用检查点

Question

我写模块有时（无法预测）获取新信息。所以我只有一个例子和一步学习。

classifier = SKCompat(tf.contrib.learn.DNNClassifier(feature_columns=feature_columns, 
                                            hidden_units=[7, 4, 2], 
                                            n_classes=2,
                                            optimizer=tf.train.GradientDescentOptimizer(0.08)
                                                    )
                     )

INFO:tensorflow:Create CheckpointSaverHook.
INFO:tensorflow:Saving checkpoints for 1 into C:\Users\IVAN~1.KAP\AppData\Local\Temp\tmprit6vryq\model.ckpt.
INFO:tensorflow:loss = 0.71007, step = 1
INFO:tensorflow:Loss for final step: 0.71007.
INFO:tensorflow:Restoring parameters from C:\Users\IVAN~1.KAP\AppData\Local\Temp\tmprit6vryq\model.ckpt-1

为每个合适的网络创建检查点。这需要3-4秒。我怎么能禁用它？这对我来说太长了

Answer 1

您可以将config参数传递给DNNClassifier，这是RunConfig的一个实例，并且有save_checkpoint_secs / save_checkpoint_steps个成员，您可以调整这些参数来更改保存的频率。