Key fix for effnetv2

oguzhanbsolak · Oct 16, 2024 · 7f77353 · 7f77353
1 parent db0e234
commit 7f77353
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/scripts/gen_effnet2_max78002.sh b/scripts/gen_effnet2_max78002.sh
@@ -3,4 +3,4 @@ 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 "$@"
+python ai8xize.py --test-dir $TARGET --prefix effnetv2_imagenet --softmax --checkpoint-file trained/ai87-imagenet-effnet2-q.pth.tar --config-file networks/ai87-imagenet-effnet2.yaml $COMMON_ARGS "$@"
diff --git a/trained/ai87-imagenet-effnet2-q.pth.tar b/trained/ai87-imagenet-effnet2-q.pth.tar
diff --git a/trained/ai87-imagenet-effnet2.pth.tar b/trained/ai87-imagenet-effnet2.pth.tar