From 67e675cdc21afc6f7827d1de6fc104c095f26b59 Mon Sep 17 00:00:00 2001
From: vanhuytran2610 <62320089+vanhuytran2610@users.noreply.github.com>
Date: Tue, 20 Apr 2021 21:43:16 +0700
Subject: [PATCH] Update generate_tfrecord.py

---
 generate_tfrecord.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/generate_tfrecord.py b/generate_tfrecord.py
index bb7379a5fb4..7936492cd11 100644
--- a/generate_tfrecord.py
+++ b/generate_tfrecord.py
@@ -15,7 +15,7 @@
 import os
 import io
 import pandas as pd
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import sys
 
 sys.path.append("../../models/research")