add cleverhans.utils_tf.silence

bjtuxck · Oct 12, 2018 · 035118a · 035118a
1 parent 95219b8
commit 035118a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/cleverhans/utils_tf.py b/cleverhans/utils_tf.py
@@ -520,3 +520,10 @@ def get_available_gpus():
   """
   local_device_protos = device_lib.list_local_devices()
   return [x.name for x in local_device_protos if x.device_type == 'GPU']
+
+
+def silence():
+  """
+  Silences tensorflaw's default printed messages
+  """
+  os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'