Skip to content

GitLab

Commit 844c1eb4 authored by Neal Wu's avatar Neal Wu
Browse files

Set tf.logging verbosity to INFO

parent d01a7239
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 5 deletions
resnet/resnet_main.py
View file @ 844c1eb4
...@@ -162,10 +162,8 @@ def evaluate(hps): ...@@ -162,10 +162,8 @@ def evaluate(hps):
tag='Best Precision', simple_value=best_precision) tag='Best Precision', simple_value=best_precision)
summary_writer.add_summary(best_precision_summ, train_step) summary_writer.add_summary(best_precision_summ, train_step)
summary_writer.add_summary(summaries, train_step) summary_writer.add_summary(summaries, train_step)
output_str = 'loss: %.3f, precision: %.3f, best precision: %.3f\n' % \ tf.logging.info('loss: %.3f, precision: %.3f, best precision: %.3f' %
(loss, precision, best_precision) (loss, precision, best_precision))
tf.logging.info(output_str)
print(output_str)
summary_writer.flush() summary_writer.flush()
if FLAGS.eval_once: if FLAGS.eval_once:
...@@ -210,4 +208,5 @@ def main(_): ...@@ -210,4 +208,5 @@ def main(_):
if __name__ == '__main__': if __name__ == '__main__':
tf.logging.set_verbosity(tf.logging.INFO)
tf.app.run() tf.app.run()
resnet/resnet_model.py
View file @ 844c1eb4
...@@ -221,7 +221,7 @@ class ResNet(object): ...@@ -221,7 +221,7 @@ class ResNet(object):
[(out_filter-in_filter)//2, (out_filter-in_filter)//2]]) [(out_filter-in_filter)//2, (out_filter-in_filter)//2]])
x += orig_x x += orig_x
tf.logging.info('image after unit %s', x.get_shape()) tf.logging.debug('image after unit %s', x.get_shape())
return x return x
def _bottleneck_residual(self, x, in_filter, out_filter, stride, def _bottleneck_residual(self, x, in_filter, out_filter, stride,
......
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