Commit d7789457 authored by Shining Sun's avatar Shining Sun Committed by Toby Boyd
Browse files

Added main function to the two keras resent benchmarks (#6546)

* Added main function to the two keras resent benchmarks

* remove asserts
parent f7e5d49a
......@@ -233,3 +233,7 @@ class Resnet56KerasBenchmarkReal(Resnet56KerasBenchmarkBase):
super(Resnet56KerasBenchmarkReal, self).__init__(
output_dir=output_dir, default_flags=default_flags)
if __name__ == '__main__':
tf.test.main()
......@@ -666,3 +666,7 @@ class TrivialKerasBenchmarkReal(keras_benchmark.KerasBenchmark):
stats,
total_batch_size=FLAGS.batch_size,
log_steps=FLAGS.log_steps)
if __name__ == '__main__':
tf.test.main()
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