KnetArray to CuArray

KnetML · Aug 7, 2023 · 7d2b437 · 7d2b437
1 parent ddd807d
commit 7d2b437
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/01-simple-mlp.ipynb b/examples/01-simple-mlp.ipynb
@@ -60,8 +60,8 @@
    "metadata": {},
    "source": [
     "The minibatch includes 2-tuples of 784x128 matrix with the flattened pixel data and a 128 vector with the teaching input; i.e. the labels in a range 1-10.    \n",
-    "If a functional GPU is detected, the array type is `KnetArray`, otherwise its a normal `Array`. \n",
-    "Computations with KnetArrays are performed on the GPU without need to care in the calling code!\n",
+    "If a functional GPU is detected, the array type is `CuArray`, otherwise its a normal `Array`. \n",
+    "Computations with CuArrays are performed on the GPU without need to care in the calling code!\n",
     "\n",
     "Data looks like:"
    ]