intel
diff --git a/‎examples/notebook/tensorflow/alexnet_mnist/conda_set_env.sh
Copy file name to clipboard
100644100755 b/‎examples/notebook/tensorflow/alexnet_mnist/conda_set_env.sh
Copy file name to clipboard
100644100755
diff --git a/‎examples/notebook/tensorflow/alexnet_mnist/devcloud_setup_env.sh
Copy file name to clipboard
100644100755 b/‎examples/notebook/tensorflow/alexnet_mnist/devcloud_setup_env.sh
Copy file name to clipboard
100644100755
diff --git a/‎examples/notebook/tensorflow/alexnet_mnist/inc_quantize_model.py
Copy file name to clipboard
+6-1Lines changed: 6 additions & 1 deletion b/‎examples/notebook/tensorflow/alexnet_mnist/inc_quantize_model.py
Copy file name to clipboard
+6-1Lines changed: 6 additions & 1 deletion
diff --git a/‎examples/notebook/tensorflow/alexnet_mnist/inc_sample_for_tensorflow.ipynb
Copy file name to clipboard
+1-11Lines changed: 1 addition & 11 deletions b/‎examples/notebook/tensorflow/alexnet_mnist/inc_sample_for_tensorflow.ipynb
Copy file name to clipboard
+1-11Lines changed: 1 addition & 11 deletions
diff --git a/‎examples/notebook/tensorflow/alexnet_mnist/pip_set_env.sh
Copy file name to clipboard
100644100755 b/‎examples/notebook/tensorflow/alexnet_mnist/pip_set_env.sh
Copy file name to clipboard
100644100755
diff --git a/‎examples/notebook/tensorflow/alexnet_mnist/requirements.txt
Copy file name to clipboard
+2-2Lines changed: 2 additions & 2 deletions b/‎examples/notebook/tensorflow/alexnet_mnist/requirements.txt
Copy file name to clipboard
+2-2Lines changed: 2 additions & 2 deletions
diff --git a/‎examples/notebook/tensorflow/alexnet_mnist/run_in_intel_devcloud.sh
Copy file name to clipboard
100644100755 b/‎examples/notebook/tensorflow/alexnet_mnist/run_in_intel_devcloud.sh
Copy file name to clipboard
100644100755
diff --git a/‎examples/notebook/tensorflow/alexnet_mnist/run_jupyter.sh
Copy file name to clipboard
100644100755 b/‎examples/notebook/tensorflow/alexnet_mnist/run_jupyter.sh
Copy file name to clipboard
100644100755
diff --git a/‎examples/notebook/tensorflow/alexnet_mnist/run_sample.sh
Copy file name to clipboard
100644100755
+1Lines changed: 1 addition & 0 deletions b/‎examples/notebook/tensorflow/alexnet_mnist/run_sample.sh
Copy file name to clipboard
100644100755
+1Lines changed: 1 addition & 0 deletions
@@ -14,6 +14,7 @@
 from neural_compressor.config import PostTrainingQuantConfig, AccuracyCriterion, TuningCriterion
 from neural_compressor.data import DataLoader
 from neural_compressor.quantization import fit
+from neural_compressor import Metric
 
 import mnist_dataset
 
@@ -58,11 +59,15 @@ def auto_tune(input_graph_path, batch_size):
                                          criterion='relative',  
                                          tolerable_loss=0.01  )
                                     )
+    top1 = Metric(name="topk", k=1)
+
     q_model = fit(
         model=input_graph_path,
         conf=config,
         calib_dataloader=dataloader,
-        eval_dataloader=dataloader)
+        eval_dataloader=dataloader,
+        eval_metric=top1
+        )
 
 
     return q_model
 
@@ -184,16 +184,6 @@
     "!./run_sample.sh"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "d53543c9",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "!ls"
-   ]
-  },
   {
    "cell_type": "markdown",
    "id": "b3cb8011-31c4-4a7c-be00-775d2ec940f4",
@@ -272,7 +262,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.10"
+   "version": "3.8.13"
   },
   "nbTranslate": {
    "displayLangs": [
 
@@ -1,5 +1,5 @@
-neural-compressor==2.0
+neural-compressor
 runipy
 notebook
 matplotlib
-tensorflow==2.12.*
+tensorflow
@@ -6,6 +6,7 @@ echo "Enable Intel Optimized TensorFlow 2.6.0 and newer by setting environment v
 echo "That will accelerate training and inference, and  it's mandatory requirement of running Intel® Neural Compressor quantize Fp32 model or deploying the quantized model."
 
 export TF_ENABLE_ONEDNN_OPTS=1
+export CUDA_VISIBLE_DEVICES=-1
 
 echo "Train Model by Keras/Tensorflow with MNIST"
 python keras_tf_train_mnist.py