Adapt Example for TF2.14 AutoTrackable API Change (#1290)

Signed-off-by: zehao-intel <[email protected]> (cherry picked from commit 9079ff1)
intel · Sep 27, 2023 · 424cf3a · 424cf3a
1 parent e7cbd0e
commit 424cf3a
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/...ject_detection/tensorflow_models/faster_rcnn_inception_resnet_v2/quantization/ptq/main.py b/...ject_detection/tensorflow_models/faster_rcnn_inception_resnet_v2/quantization/ptq/main.py
@@ -53,7 +53,7 @@ def evaluate(model):
         accuracy (float): evaluation result, the larger is better.
     """
     from neural_compressor.model import Model
-    from tensorflow.python.training.tracking.tracking import AutoTrackable  
+    from tensorflow.python.trackable.autotrackable import AutoTrackable
     warmup = 5      
     iteration = -1
     if args.benchmark and args.mode == 'performance':

diff --git a/...sorflow/object_detection/tensorflow_models/faster_rcnn_resnet101/quantization/ptq/main.py b/...sorflow/object_detection/tensorflow_models/faster_rcnn_resnet101/quantization/ptq/main.py
@@ -53,7 +53,7 @@ def evaluate(model):
         accuracy (float): evaluation result, the larger is better.
     """
     from neural_compressor.model import Model
-    from tensorflow.python.training.tracking.tracking import AutoTrackable  
+    from tensorflow.python.trackable.autotrackable import AutoTrackable
     warmup = 5      
     iteration = -1
     if args.benchmark and args.mode == 'performance':

diff --git a/neural_compressor/adaptor/tf_utils/graph_converter.py b/neural_compressor/adaptor/tf_utils/graph_converter.py
@@ -83,7 +83,7 @@
     version1_lte_version2,
 )
 
-TF_SUPPORTED_MAX_VERSION = "2.13.0"
+TF_SUPPORTED_MAX_VERSION = "2.14.0"
 TF_SUPPORTED_MIN_VERSION = "1.14.0"
 
 logger = logging.getLogger("neural_compressor")

diff --git a/neural_compressor/adaptor/tf_utils/graph_converter_without_calib.py b/neural_compressor/adaptor/tf_utils/graph_converter_without_calib.py
@@ -56,7 +56,7 @@
     version1_lte_version2,
 )
 
-TF_SUPPORTED_MAX_VERSION = "2.13.0"
+TF_SUPPORTED_MAX_VERSION = "2.14.0"
 TF_SUPPORTED_MIN_VERSION = "1.14.0"
 
 logger = logging.getLogger("neural_compressor")