在Tensorflow中的DNNClassifier估算器中记录设备信息

Question

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

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

主要问题：我的3分层神经网络与隐藏单位[100, 75, 50]在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)))