tensorflow - 在 Tensorflow 的 DNNClassifier 估计器中记录设备信息

Question

我正在使用 DNNClassifier Estimator 来训练二元分类器。我想记录设备信息以验证我的模型是在 GPU 还是 CPU 上运行。

由于使用 Estimator 我们不处理 session，我该如何记录设备信息？

主要问题:我的带有隐藏单元 [100, 75, 50] 的 3 分层神经网络在 CPU 上的运行速度比 GPU 快。我试图将批量大小增加到 256，但仍然相同。因此，我想确认它是否真的在使用 GPU。

Answer 1

使用 tf.estimator.Estimator.__init__ 的 config 参数:

classifier = \
  DNNClassifier(feature_columns=feature_columns,
                hidden_units=[100, 75, 50],
                config=tf.estimator.RunConfig(session_config=tf.ConfigProto(log_device_placement=True)))