Remove logging which generates tons of logs for large model.
PiperOrigin-RevId: 211707155
This commit is contained in:
parent
7e2577b098
commit
59c43f26de
@ -30,7 +30,6 @@ from tensorflow.python.framework import ops
|
|||||||
from tensorflow.python.ops import control_flow_ops
|
from tensorflow.python.ops import control_flow_ops
|
||||||
from tensorflow.python.ops import gen_resource_variable_ops
|
from tensorflow.python.ops import gen_resource_variable_ops
|
||||||
from tensorflow.python.ops import variable_scope
|
from tensorflow.python.ops import variable_scope
|
||||||
from tensorflow.python.platform import tf_logging as logging
|
|
||||||
|
|
||||||
|
|
||||||
@contextlib.contextmanager
|
@contextlib.contextmanager
|
||||||
@ -258,7 +257,6 @@ def replicated_scope(num_replicas):
|
|||||||
collections = [ops.GraphKeys.GLOBAL_VARIABLES]
|
collections = [ops.GraphKeys.GLOBAL_VARIABLES]
|
||||||
kwargs["collections"] = []
|
kwargs["collections"] = []
|
||||||
|
|
||||||
logging.info("Constructing replicated variable %s", name)
|
|
||||||
variables = []
|
variables = []
|
||||||
index = {}
|
index = {}
|
||||||
for i in range(num_replicas):
|
for i in range(num_replicas):
|
||||||
|
Loading…
Reference in New Issue
Block a user