在keras.utils.multi_gpu_model中关闭调试信息

时间:2018-07-20 21:03:50

标签: tensorflow keras gpu trace

从keras.utils导入multi_gpu_model
    模型= multi_gpu_model(model,gpus = ngpus)

如何关闭以下调试信息?

即使从硬盘加载保存的模型,也会发生这种情况。

非常感谢!

=============
turn off such outputs: 

batch_normalization_4/AssignMovingAvg_1/batch_normalization_4/moving_variance/sub_2/x: (Const): /job:localhost/replica:0/task:0/device:GPU:0
2018-07-20 16:53:03.520949: I tensorflow/core/common_runtime/placer.cc:886] batch_normalization_4/AssignMovingAvg_1/batch_normalization_4/moving_variance/sub_2/x: (Const)/job:localhost/replica:0/task:0/device:GPU:0
batch_normalization_4/AssignMovingAvg_1/batch_normalization_4/moving_variance/sub_1/x: (Const): /job:localhost/replica:0/task:0/device:GPU:0
2018-07-20 16:53:03.520961: I tensorflow/core/common_runtime/placer.cc:886] batch_normalization_4/AssignMovingAvg_1/batch_normalization_4/moving_variance/sub_1/x: (Const)/job:localhost/replica:0/task:0/device:GPU:0


==============
Versions:
tensorboard==1.8.0
tensorflow==1.3.0
tensorflow-gpu==1.8.0
tensorflow-tensorboard==0.1.8
Keras==2.2.0
Keras-Applications==1.0.2
Keras-Preprocessing==1.0.1

0 个答案:

没有答案