Fix Quickstart Example on Readme (#1082)

Signed-off-by: zehao-intel <[email protected]> (cherry picked from commit 1276f0f)
intel · Jul 21, 2023 · a7721fc · a7721fc
1 parent 8af7437
commit a7721fc
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 21 deletions.
diff --git a/README.md b/README.md
@@ -45,9 +45,8 @@ pip install tensorflow
 wget https://storage.googleapis.com/intel-optimized-tensorflow/models/v1_6/mobilenet_v1_1.0_224_frozen.pb
 ```
 ```python
+from neural_compressor.data import DataLoader, Datasets
 from neural_compressor.config import PostTrainingQuantConfig
-from neural_compressor.data import DataLoader
-from neural_compressor.data import Datasets
 
 dataset = Datasets('tensorflow')['dummy'](shape=(1, 224, 224, 3))
 dataloader = DataLoader(framework='tensorflow', dataset=dataset)
@@ -56,8 +55,7 @@ from neural_compressor.quantization import fit
 q_model = fit(
     model="./mobilenet_v1_1.0_224_frozen.pb",
     conf=PostTrainingQuantConfig(),
-    calib_dataloader=dataloader,
-    eval_dataloader=dataloader)
+    calib_dataloader=dataloader)
 ```
 
 ## Documentation

diff --git a/docs/source/get_started.md b/docs/source/get_started.md
@@ -15,20 +15,17 @@ pip install tensorflow
 wget https://storage.googleapis.com/intel-optimized-tensorflow/models/v1_6/mobilenet_v1_1.0_224_frozen.pb
 ```
 ```python
+from neural_compressor.data import DataLoader, Datasets
 from neural_compressor.config import PostTrainingQuantConfig
-from neural_compressor.data import DataLoader
-from neural_compressor.data import Datasets
 
 dataset = Datasets('tensorflow')['dummy'](shape=(1, 224, 224, 3))
 dataloader = DataLoader(framework='tensorflow', dataset=dataset)
 
 from neural_compressor.quantization import fit
-config = PostTrainingQuantConfig()
 q_model = fit(
     model="./mobilenet_v1_1.0_224_frozen.pb",
-    conf=config,
-    calib_dataloader=dataloader,
-    eval_dataloader=dataloader)
+    conf=PostTrainingQuantConfig(),
+    calib_dataloader=dataloader)
 ```
 
 ## Validated Models

diff --git a/examples/helloworld/tf_example7/README.md b/examples/helloworld/tf_example7/README.md
@@ -32,12 +32,8 @@ We can quantize a model only needing to set the dataloader with dummy dataset to
     dataloader = DataLoader(framework='tensorflow', dataset=dataset)
     # Post Training Quantization Config
     config = PostTrainingQuantConfig()
-    # Built-in topk metric
-    top1 = Metric(name="topk", k=1)
     # Just call fit to do quantization.
     q_model = fit(model="./mobilenet_v1_1.0_224_frozen.pb",
                   conf=config,
-                  calib_dataloader=dataloader,
-                  eval_dataloader=dataloader,
-                  eval_metric=top1)
+                  calib_dataloader=dataloader)
 ```
diff --git a/examples/helloworld/tf_example7/test.py b/examples/helloworld/tf_example7/test.py
@@ -1,4 +1,3 @@
-from neural_compressor import Metric
 from neural_compressor.config import PostTrainingQuantConfig
 from neural_compressor.data import DataLoader
 from neural_compressor.data import Datasets
@@ -13,14 +12,10 @@ def main():
     dataloader = DataLoader(framework='tensorflow', dataset=dataset)
     # Post Training Quantization Config
     config = PostTrainingQuantConfig()
-    # Built-in topk metric
-    top1 = Metric(name="topk", k=1)
     # Just call fit to do quantization.
     q_model = fit(model="./mobilenet_v1_1.0_224_frozen.pb",
                   conf=config,
-                  calib_dataloader=dataloader,
-                  eval_dataloader=dataloader,
-                  eval_metric=top1)
+                  calib_dataloader=dataloader)
 
 
 if __name__ == "__main__":