从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