chore: update cifar finetuning

zama-ai · Sep 23, 2024 · e57809c · e57809c
1 parent 9b1c77f
commit e57809c
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 9 deletions.
diff --git a/use_case_examples/cifar/cifar_brevitas_finetuning/CifarInFhe.ipynb b/use_case_examples/cifar/cifar_brevitas_finetuning/CifarInFhe.ipynb
@@ -63,8 +63,10 @@
     "bit = 5\n",
     "seed = 42\n",
     "\n",
-    "use_gpu_if_available = False\n",
-    "device = \"cuda\" if use_gpu_if_available and check_gpu_available() else \"cpu\"\n",
+    "# To use the CUDA backend for the Concrete compiler, install the GPU-supported version.\n",
+    "# Refer to the official guide:\n",
+    "# https://github.com/zama-ai/concrete-ml/blob/main/docs/guides/using_gpu.md\n",
+    "device = \"cuda\" if check_gpu_available() else \"cpu\"\n",
     "\n",
     "print(f\"Device Type: {device}\")"
    ]

diff --git a/use_case_examples/cifar/cifar_brevitas_finetuning/CifarInFheWithSmallerAccumulators.ipynb b/use_case_examples/cifar/cifar_brevitas_finetuning/CifarInFheWithSmallerAccumulators.ipynb
@@ -61,8 +61,10 @@
     "seed = 42\n",
     "rounding_thresholds_bits = [8, 7, 6, 5, 3]\n",
     "\n",
-    "use_gpu_if_available = False\n",
-    "device = \"cuda\" if use_gpu_if_available and check_gpu_available() else \"cpu\""
+    "# To use the CUDA backend for the Concrete compiler, install the GPU-supported version.\n",
+    "# Refer to the official guide:\n",
+    "# https://github.com/zama-ai/concrete-ml/blob/main/docs/guides/using_gpu.md\n",
+    "device = \"cuda\" if check_gpu_available() else \"cpu\""
    ]
   },
   {

diff --git a/use_case_examples/cifar/cifar_brevitas_finetuning/CifarQuantizationAwareTraining.ipynb b/use_case_examples/cifar/cifar_brevitas_finetuning/CifarQuantizationAwareTraining.ipynb
@@ -89,8 +89,10 @@
    "source": [
     "bit = 5\n",
     "\n",
-    "use_gpu_if_available = False\n",
-    "device = \"cuda\" if use_gpu_if_available and check_gpu_available() else \"cpu\"\n",
+    "# To use the CUDA backend for the Concrete compiler, install the GPU-supported version.\n",
+    "# Refer to the official guide:\n",
+    "# https://github.com/zama-ai/concrete-ml/blob/main/docs/guides/using_gpu.md\n",
+    "device = \"cuda\" if check_gpu_available() else \"cpu\"\n",
     "\n",
     "print(f\"Device Type: {device}\")"
    ]

diff --git a/use_case_examples/cifar/cifar_brevitas_finetuning/FromImageNetToCifar.ipynb b/use_case_examples/cifar/cifar_brevitas_finetuning/FromImageNetToCifar.ipynb
@@ -65,8 +65,10 @@
    "source": [
     "dataset_name = \"CIFAR_100\"\n",
     "\n",
-    "use_gpu_if_available = False\n",
-    "device = \"cuda\" if use_gpu_if_available and check_gpu_available() else \"cpu\"\n",
+    "# To use the CUDA backend for the Concrete compiler, install the GPU-supported version.\n",
+    "# Refer to the official guide:\n",
+    "# https://github.com/zama-ai/concrete-ml/blob/main/docs/guides/using_gpu.md#usage-preqreuisites\n",
+    "device = \"cuda\" if check_gpu_available() else \"cpu\"\n",
     "\n",
     "param_c10 = {\n",
     "    \"output_size\": 10,\n",

diff --git a/use_case_examples/cifar/cifar_brevitas_finetuning/cifar_utils.py b/use_case_examples/cifar/cifar_brevitas_finetuning/cifar_utils.py
@@ -457,7 +457,7 @@ def fhe_compatibility(model: Callable, data: DataLoader, device: str) -> Callabl
         torch_inputset=data,
         show_mlir=False,
         output_onnx_file="test.onnx",
-        configuration=Configuration(use_gpu=(device == "cuda")),
+        device=device,
     )
 
     return qmodel