diff --git a/scripts/gen_effnet2_max78002.sh b/scripts/gen_effnet2_max78002.sh
index d906a98b..595f4059 100755
--- 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 "$@"
\ No newline at end of file
+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 "$@"
\ 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 0af06e28..5f080c79 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 ed2f56e7..03898180 100644
Binary files a/trained/ai87-imagenet-effnet2.pth.tar and b/trained/ai87-imagenet-effnet2.pth.tar differ