Fixed ofirgo-san's review comment.[3]

sony · Jan 31, 2025 · 67b78b5 · 67b78b5
1 parent 8128264
commit 67b78b5
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/...otebooks/mct_features_notebooks/pytorch/example_pytorch_activation_threshold_search.ipynb b/...otebooks/mct_features_notebooks/pytorch/example_pytorch_activation_threshold_search.ipynb
@@ -506,11 +506,14 @@
    "source": [
     "import matplotlib.pyplot as plt\n",
     "\n",
+    "ls_table = ['--', ':']\n",
+    "lc_table = ['blue', 'red']\n",
+    "\n",
     "# Plotting\n",
     "plt.figure(figsize=(10, 6))\n",
     "plt.hist(all_activations_relu, bins=100, alpha=0.5, label='Original')\n",
-    "for method, threshold in optimal_thresholds_relu.items():\n",
-    "    plt.axvline(threshold, linestyle='--', linewidth=2, label=f'{method}: {threshold:.2f}')\n",
+    "for index, (method, threshold) in enumerate(optimal_thresholds_relu.items()):\n",
+    "    plt.axvline(threshold, linestyle=ls_table[index], color=lc_table[index], linewidth=2, label=f'{method}: {threshold:.2f}')\n",
     "\n",
     "plt.title('Activation Distribution with Optimal Quantization Thresholds First Relu Layer')\n",
     "plt.xlabel('Activation Value')\n",
@@ -530,11 +533,14 @@
    "source": [
     "import matplotlib.pyplot as plt\n",
     "\n",
+    "ls_table = ['--', ':']\n",
+    "lc_table = ['blue', 'red']\n",
+    "\n",
     "# Plotting\n",
     "plt.figure(figsize=(10, 6))\n",
     "plt.hist(all_activations_project, bins=100, alpha=0.5, label='Original')\n",
-    "for method, threshold in optimal_thresholds_project.items():\n",
-    "    plt.axvline(threshold, linestyle='--', linewidth=2, label=f'{method}: {threshold:.2f}')\n",
+    "for index, (method, threshold) in enumerate(optimal_thresholds_project.items()):\n",
+    "    plt.axvline(threshold, linestyle=ls_table[index], color=lc_table[index], linewidth=2, label=f'{method}: {threshold:.2f}')\n",
     "\n",
     "plt.title('Activation Distribution with Optimal Quantization Thresholds Project BN layer')\n",
     "plt.xlabel('Activation Value')\n",
@@ -756,7 +762,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.5"
+   "version": "3.10.16"
   }
  },
  "nbformat": 4,