diff --git a/scripts/gen_autoencoder_voyager4_max78000.sh b/scripts/gen_autoencoder_voyager4_max78000.sh new file mode 100755 index 00000000..b1388462 --- /dev/null +++ b/scripts/gen_autoencoder_voyager4_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_voyager4 --checkpoint-file trained/ai85-autoencoder-motordatavoyager4v1-qat-q.pth.tar --config-file networks/ai85-autoencoder.yaml --sample-input tests/sample_motordatavoyager4_fortrain.npy --energy $COMMON_ARGS "$@" diff --git a/scripts/quantize_autoencoder_voyager4_max78000.sh b/scripts/quantize_autoencoder_voyager4_max78000.sh new file mode 100755 index 00000000..4585491b --- /dev/null +++ b/scripts/quantize_autoencoder_voyager4_max78000.sh @@ -0,0 +1,3 @@ +#!/bin/sh +python quantize.py trained/ai85-autoencoder-motordatavoyager4-qat.pth.tar trained/ai85-autoencoder-motordatavoyager4-qat-q.pth.tar --device MAX78000 "$@" + diff --git a/tests/sample_motordatavoyager4_fortrain.npy b/tests/sample_motordatavoyager4_fortrain.npy new file mode 100644 index 00000000..fe809805 Binary files /dev/null and b/tests/sample_motordatavoyager4_fortrain.npy differ diff --git a/trained/ai85-autoencoder-motordatavoyager4-qat-q.pth.tar b/trained/ai85-autoencoder-motordatavoyager4-qat-q.pth.tar new file mode 100644 index 00000000..f8f2b81e Binary files /dev/null and b/trained/ai85-autoencoder-motordatavoyager4-qat-q.pth.tar differ diff --git a/trained/ai85-autoencoder-motordatavoyager4-qat.pth.tar b/trained/ai85-autoencoder-motordatavoyager4-qat.pth.tar new file mode 100644 index 00000000..ffb21485 Binary files /dev/null and b/trained/ai85-autoencoder-motordatavoyager4-qat.pth.tar differ