Add Autoencoder Model

networks/ai85-autoencoder.yaml

@@ -0,0 +1,47 @@
+---
+arch: ai85autoencoder
+dataset: SampleMotorDataLimerick
+# Define layer parameters in order of the layer sequence
+layers:
+  # Layer 0: 256 channels in --> 64 processors, 4 passes. Conv1D
+  - pad: 0
+    activate: ReLU
+    out_offset: 0x2000
+    processors: 0xffffffffffffffff
+    operation: Conv1d
+    kernel_size: 1
+    data_format: HWC
+  # Layer 1: 128 channels in --> 64 processors, 2 passes. Conv1D
+  - pad: 0
+    activate: ReLU
+    out_offset: 0x0000
+    processors: 0xffffffffffffffff
+    kernel_size: 3
+    operation: Conv1d
+  # Layer 2: 64 inputs --> 64 processors, 1 passes. MLP
+  - pad: 0
+    activate: ReLU
+    out_offset: 0x2000
+    processors: 0xffffffffffffffff
+    operation: MLP
+  # Layer 3: 32 inputs --> 32 processors, 1 pass. MLP
+  - activate: ReLU
+    out_offset: 0x0000
+    processors: 0x00000000ffffffff
+    operation: MLP
+  # Layer 4: 16 inputs --> 16 processors, 1 pass. MLP
+  - activate: ReLU
+    out_offset: 0x2000
+    processors: 0x000000000000000f
+    operation: MLP
+  # Layer 5: 32 inputs --> 32 processors, 1 pass. MLP
+  - activate: ReLU
+    out_offset: 0x4000
+    processors: 0x00000000ffffffff
+    operation: MLP
+  # Layer 6: 96 inputs --> 48 processors, 2 passes. MLP
+  - pad: 0
+    activate: None
+    out_offset: 0x0000
+    processors: 0x0000ffffffffffff
+    operation: MLP

scripts/gen_autoencoder_max78000.sh

@@ -0,0 +1,6 @@
+#!/bin/sh
+DEVICE="MAX78000"
+TARGET="sdk/Examples/$DEVICE/CNN"
+COMMON_ARGS="--device $DEVICE --timer 0 --display-checkpoint --verbose"
+
+python ai8xize.py --test-dir $TARGET --prefix autoencoder --checkpoint-file trained/ai85-autoencoder-samplemotordatalimerick-qat-q.pth.tar --config-file networks/ai85-autoencoder.yaml --sample-input tests/sample_motordatalimerick_fortrain.npy --energy  $COMMON_ARGS "$@"

scripts/quantize_autoencoder_max78000.sh

@@ -0,0 +1,2 @@
+#!/bin/sh
+python quantize.py trained/ai85-autoencoder-samplemotordatalimerick-qat.pth.tar trained/ai85-autoencoder-samplemotordatalimerick-qat-q.pth.tar --device MAX78000 "$@"