Update base image, and add parameters file

Dockerfile

@@ -3,7 +3,7 @@ ARG UBUNTU_VERSION=20.04
 
 ARG ARCH=
 ARG CUDA=11.2
-FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}.1-base-ubuntu${UBUNTU_VERSION} as base
+FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}.2-base-ubuntu${UBUNTU_VERSION} as base
 ARG CUDA
 ARG CUDNN=8.1.0.77-1
 ARG CUDNN_MAJOR_VERSION=8
@@ -73,4 +73,6 @@ WORKDIR /scripts
 # Copy the normal files (e.g. run.sh and the extract_dataset scripts in)
 COPY . ./
 
+ENV PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
+
 ENTRYPOINT ["/bin/bash", "run.sh"]

README.md

@@ -94,4 +94,4 @@ You can also push this block back to Edge Impulse, that makes it available like
     $ edge-impulse-blocks push
     ```
 
-2. The block is now available under any of your projects, via  **Create impulse > Add learning block > Object Detection (Images)**.
+2. The block is now available under any of your projects, via  **Create impulse > Add learning block**.

parameters.json

@@ -0,0 +1,7 @@
+[{
+    "name": "Number of training cycles",
+    "value": "30",
+    "type": "int",
+    "help": "Number of epochs to train the neural network on.",
+    "param": "epochs"
+}]

run.sh

@@ -14,11 +14,6 @@ while [[ $# -gt 0 ]]; do
       shift # past argument
       shift # past value
       ;;
-    --learning-rate) # e.g. 0.01
-      LEARNING_RATE="$2"
-      shift # past argument
-      shift # past value
-      ;;
     --data-directory) # e.g. 0.2
       DATA_DIRECTORY="$2"
       shift # past argument
@@ -40,10 +35,6 @@ if [ -z "$EPOCHS" ]; then
     echo "Missing --epochs"
     exit 1
 fi
-if [ -z "$LEARNING_RATE" ]; then
-    echo "Missing --learning-rate"
-    exit 1
-fi
 if [ -z "$DATA_DIRECTORY" ]; then
     echo "Missing --data-directory"
     exit 1