Skip to content

GitLab

Commit 0fa5ff23 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 265771605
parent 1f5a5e9d
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
official/transformer/v2/transformer_main.py
View file @ 0fa5ff23
...@@ -25,7 +25,7 @@ from __future__ import print_function ...@@ -25,7 +25,7 @@ from __future__ import print_function
import os import os
import tempfile import tempfile
from absl import app as absl_app # pylint: disable=unused-import from absl import app
from absl import flags from absl import flags
from absl import logging from absl import logging
import tensorflow as tf import tensorflow as tf
...@@ -454,4 +454,4 @@ def main(_): ...@@ -454,4 +454,4 @@ def main(_):
if __name__ == "__main__": if __name__ == "__main__":
logging.set_verbosity(logging.INFO) logging.set_verbosity(logging.INFO)
misc.define_transformer_flags() misc.define_transformer_flags()
absl_app.run(main) app.run(main)
official/vision/image_classification/resnet_imagenet_main.py
View file @ 0fa5ff23
...@@ -18,7 +18,7 @@ from __future__ import absolute_import ...@@ -18,7 +18,7 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
from absl import app as absl_app from absl import app
from absl import flags from absl import flags
from absl import logging from absl import logging
import tensorflow as tf import tensorflow as tf
...@@ -282,4 +282,4 @@ def main(_): ...@@ -282,4 +282,4 @@ def main(_):
if __name__ == '__main__': if __name__ == '__main__':
logging.set_verbosity(logging.INFO) logging.set_verbosity(logging.INFO)
define_imagenet_keras_flags() define_imagenet_keras_flags()
absl_app.run(main) app.run(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