Update generate_tfrecord.py

jackman815 · Aug 12, 2021 · ef0d566 · ef0d566
1 parent 1833c01
commit ef0d566
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/generate_tfrecord.py b/generate_tfrecord.py
@@ -17,6 +17,7 @@
 import pandas as pd
 import tensorflow as tf
 import sys
+import absl
 
 sys.path.append("../../models/research")
 
@@ -131,4 +132,4 @@ def main(_):
 
 
 if __name__ == "__main__":
-    tf.app.run()
+    absl.run()