Python disable_layer_logging示例

示例#1

文件： concurrency.py 项目： quanlzheng/tensorpack

 def _init_runtime(self):
     """ Call _init_runtime under different CUDA_VISIBLE_DEVICES, you'll
         have workers that run on multiGPUs
     """
     if self.idx != 0:
         from tensorpack.models.registry import disable_layer_logging
         disable_layer_logging()
     self.predictor = OfflinePredictor(self.config)
     if self.idx == 0:
         with self.predictor.graph.as_default():
             describe_trainable_vars()

示例#2

 def _init_runtime(self):
     """ Call _init_runtime under different CUDA_VISIBLE_DEVICES, you'll
         have workers that run on multiGPUs
     """
     if self.idx != 0:
         from tensorpack.models.registry import disable_layer_logging
         disable_layer_logging()
     self.predictor = OfflinePredictor(self.config)
     if self.idx == 0:
         with self.predictor.graph.as_default():
             describe_trainable_vars()