Commit 9bf59445 authored by Yuxin Wu's avatar Yuxin Wu

Simplify messages of filternonegrad

parent 2388f4b8
...@@ -64,12 +64,15 @@ class FilterNoneGrad(GradientProcessor): ...@@ -64,12 +64,15 @@ class FilterNoneGrad(GradientProcessor):
def _process(self, grads): def _process(self, grads):
g = [] g = []
to_print = []
for grad, var in grads: for grad, var in grads:
if grad is None: if grad is None:
if self._verbose: to_print.append(var.op.name)
logger.warn("No Gradient w.r.t {}".format(var.op.name))
else: else:
g.append((grad, var)) g.append((grad, var))
if self._verbose:
message = ', '.join(to_print)
logger.warn("No gradient w.r.t these trainable variables: {}".format(message))
return g return g
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment