Skip to content

GitLab

Commit d0f21f25 authored by George Karpenkov's avatar George Karpenkov Committed by A. Unique TensorFlower
Browse files

Port shakespeare benchmark to TF2 

PiperOrigin-RevId: 273527676
parent 3fa65606
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/staging/shakespeare/shakespeare_main.py
View file @ d0f21f25
...@@ -22,7 +22,7 @@ import functools ...@@ -22,7 +22,7 @@ import functools
import os import os
# pylint: disable=wrong-import-order # pylint: disable=wrong-import-order
from absl import app as absl_app from absl import app
from absl import flags from absl import flags
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
...@@ -298,4 +298,4 @@ def main(_): ...@@ -298,4 +298,4 @@ def main(_):
if __name__ == '__main__': if __name__ == '__main__':
define_flags() define_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