diff --git a/gen-demos-max78002.sh b/gen-demos-max78002.sh index a7e24576..e2ca9801 100755 --- a/gen-demos-max78002.sh +++ b/gen-demos-max78002.sh @@ -19,7 +19,7 @@ python ai8xize.py --test-dir $TARGET --prefix aisegment_unet --checkpoint-file t python ai8xize.py --test-dir $TARGET --prefix svhn_tinierssd --checkpoint-file trained/ai85-svhn-tinierssd-qat8-q.pth.tar --config-file networks/svhn-tinierssd.yaml --overlap-data $COMMON_ARGS "$@" python ai8xize.py --test-dir $TARGET --prefix cifar-100-effnet2 --checkpoint-file trained/ai87-cifar100-effnet2-qat8-q.pth.tar --config-file networks/ai87-cifar100-effnet2.yaml --softmax $COMMON_ARGS "$@" python ai8xize.py --test-dir $TARGET --prefix cifar-100-mobilenet-v2-0.75 --checkpoint-file trained/ai87-cifar100-mobilenet-v2-0.75-qat8-q.pth.tar --config-file networks/ai87-cifar100-mobilenet-v2-0.75.yaml --softmax $COMMON_ARGS "$@" -python ai8xize.py --test-dir $TARGET --prefix imagenet --checkpoint-file trained/ai87-imagenet-effnet2-q.pth.tar --config-file networks/ai87-imagenet-effnet2.yaml $COMMON_ARGS "$@" +python ai8xize.py --test-dir $TARGET --prefix effnetv2_imagenet --scale-output --softmax --checkpoint-file trained/ai87-imagenet-effnet2-q.pth.tar --config-file networks/ai87-imagenet-effnet2.yaml $COMMON_ARGS "$@" python ai8xize.py --test-dir $TARGET --prefix facedet_tinierssd --checkpoint-file trained/ai87-facedet-tinierssd-qat8-q.pth.tar --config-file networks/ai87-facedet-tinierssd.yaml --sample-input tests/sample_vggface2_facedetection.npy $COMMON_ARGS "$@" python ai8xize.py --test-dir $TARGET --prefix pascalvoc_fpndetector --checkpoint-file trained/ai87-pascalvoc-fpndetector-qat8-q.pth.tar --config-file networks/ai87-pascalvoc-fpndetector.yaml --fifo --sample-input tests/sample_pascalvoc_256_320.npy --no-unload $COMMON_ARGS "$@" python ai8xize.py --test-dir $TARGET --prefix kinetics --checkpoint-file trained/ai85-kinetics-qat8-q.pth.tar --config-file networks/ai85-kinetics-actiontcn.yaml --overlap-data --softmax --zero-sram $COMMON_ARGS "$@" diff --git a/scripts/gen_effnet2_max78002.sh b/scripts/gen_effnet2_max78002.sh new file mode 100644 index 00000000..d906a98b --- /dev/null +++ b/scripts/gen_effnet2_max78002.sh @@ -0,0 +1,6 @@ +#!/bin/sh +DEVICE="MAX78002" +TARGET="sdk/Examples/$DEVICE/CNN" +COMMON_ARGS="--device $DEVICE --timer 0 --display-checkpoint --verbose" + +python ai8xize.py --test-dir $TARGET --prefix effnetv2_imagenet --scale-output --softmax --checkpoint-file trained/ai87-imagenet-effnet2-q.pth.tar --config-file networks/ai87-imagenet-effnet2.yaml $COMMON_ARGS "$@" \ No newline at end of file diff --git a/trained/ai87-imagenet-effnet2-q.pth.tar b/trained/ai87-imagenet-effnet2-q.pth.tar index 38884f71..8418ad3e 100644 Binary files a/trained/ai87-imagenet-effnet2-q.pth.tar and b/trained/ai87-imagenet-effnet2-q.pth.tar differ diff --git a/trained/ai87-imagenet-effnet2.pth.tar b/trained/ai87-imagenet-effnet2.pth.tar index e8c8fc5e..1c55c8cb 100644 Binary files a/trained/ai87-imagenet-effnet2.pth.tar and b/trained/ai87-imagenet-effnet2.pth.tar differ