Skip to content

Commit 3fc6988

Browse files
ermanokermanok
authored
Merge branch 'analogdevicesinc:develop' into develop
2 parents cf62088 + 32aea7f commit 3fc6988

File tree

62 files changed

+24268
-300
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

62 files changed

+24268
-300
lines changed

‎.flake8

-2
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,2 @@
11
[flake8]
22
max-line-length = 99
3-
per-file-ignores =
4-
range_linear_ai84.py:E501,E128

‎.gitignore

+1-2
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
*.mem
33
*.prof
44
/.mypy_cache/
5-
/.python-version/
5+
/.venv/
66
/.vscode/
77
**/build/
88
/ninja-python-distributions
@@ -13,5 +13,4 @@
1313
/super-linter.log
1414
/super-linter.report/
1515
/.version-check
16-
/venv*/
1716
/sdk/

‎.isort.cfg

-6
This file was deleted.

‎.pylintrc

-21
This file was deleted.

‎.python-version

+1-1
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
3.8.11
1+
3.11.9

‎KNOWN_ISSUES.txt

+4
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
KNOWN ISSUES for ai8x-synthesis with PyTorch 2.3.1
2+
==================================================
3+
4+
N/A

‎README.md

+182-140
Large diffs are not rendered by default.

‎README.pdf

-2.29 MB
Binary file not shown.

‎ai8x-synthesis.code-workspace

-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
],
77
"settings": {
88
"markdown.extension.toc.levels": "2..6",
9-
"python.pythonPath": "venv/bin/python",
109
"git.ignoreLimitWarning": true,
1110
"pylint.args": ["--rcfile=.pylintrc"]
1211
},

‎assets/embedded-riscv-ai85/templateMakefile.RISCV

+4-2
Original file line numberDiff line numberDiff line change
@@ -97,8 +97,10 @@ STACK_SIZE=1024
9797
# "A2" in ASCII
9898
# TARGET_REV=0x4132
9999

100-
# Use this variables to specify and alternate tool path
101-
#TOOL_DIR=/usr/local/riscv-none-embed-gcc/8.3.0-1.1
100+
# Use this variables to specify an alternate RISCV toolchain
101+
# For example, to use the (now deprecated) "riscv-none-embed"...
102+
# Note the toolchain must be on the PATH.
103+
#RISCV_PREFIX = riscv-none-embed
102104

103105
# Point this variable to a startup file to override the default file
104106
#STARTUPFILE=startup_riscv_##__TARGET_LC__##.S

‎assets/embedded-riscv-ai87/templateMakefile.RISCV

+4-2
Original file line numberDiff line numberDiff line change
@@ -97,8 +97,10 @@ STACK_SIZE=1024
9797
# "A2" in ASCII
9898
# TARGET_REV=0x4132
9999

100-
# Use this variables to specify and alternate tool path
101-
#TOOL_DIR=/usr/local/riscv-none-embed-gcc/8.3.0-1.1
100+
# Use this variables to specify an alternate RISCV toolchain
101+
# For example, to use the (now deprecated) "riscv-none-embed"...
102+
# Note the toolchain must be on the PATH.
103+
#RISCV_PREFIX = riscv-none-embed
102104

103105
# Point this variable to a startup file to override the default file
104106
#STARTUPFILE=startup_riscv_##__TARGET_LC__##.S

‎assets/vscode/defaults/settings.json

+13-7
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,16 @@
11
{
22
"terminal.integrated.env.windows": {
3-
"Path":"${config:OCD_path};${config:ARM_GCC_path}/bin;${config:xPack_GCC_path}/bin;${config:Make_path};${env:PATH}",
3+
"Path":"${config:OCD_path};${config:ARM_GCC_path}/bin;${config:xPack_GCC_path}/bin;${config:MSYS_path}/usr/bin;${config:Make_path};${env:PATH}",
44
"MAXIM_PATH":"${config:MAXIM_PATH}"
55
},
66
"terminal.integrated.defaultProfile.windows": "Command Prompt",
77

88
"terminal.integrated.env.linux": {
9-
"PATH":"${config:OCD_path}:${config:ARM_GCC_path}/bin:${config:xPack_GCC_path}/bin:${env:PATH}",
9+
"PATH":"${config:OCD_path}:${config:ARM_GCC_path}/bin:${config:xPack_GCC_path}/bin:${config:Make_path}:${env:PATH}",
1010
"MAXIM_PATH":"${config:MAXIM_PATH}"
1111
},
1212
"terminal.integrated.env.osx": {
13-
"PATH":"${config:OCD_path}/bin:${config:ARM_GCC_path}/bin:${config:xPack_GCC_path}/bin:${env:PATH}",
13+
"PATH":"${config:OCD_path}/bin:${config:ARM_GCC_path}/bin:${config:xPack_GCC_path}/bin:${config:Make_path}:${env:PATH}",
1414
"MAXIM_PATH":"${config:MAXIM_PATH}"
1515
},
1616

@@ -28,12 +28,13 @@
2828
"RV_OCD_target_file":"${config:target}_riscv.cfg",
2929

3030
"v_Arm_GCC":"10.3",
31-
"v_xPack_GCC":"10.2.0-1.2",
31+
"v_xPack_GCC":"12.2.0-3.1",
3232

3333
"OCD_path":"${config:MAXIM_PATH}/Tools/OpenOCD",
3434
"ARM_GCC_path":"${config:MAXIM_PATH}/Tools/GNUTools/${config:v_Arm_GCC}",
35-
"xPack_GCC_path":"${config:MAXIM_PATH}/Tools/xPack/riscv-none-embed-gcc/${config:v_xPack_GCC}",
36-
"Make_path":"${config:MAXIM_PATH}/Tools/MSYS2/usr/bin",
35+
"xPack_GCC_path":"${config:MAXIM_PATH}/Tools/xPack/riscv-none-elf-gcc/${config:v_xPack_GCC}",
36+
"Make_path":"${config:MAXIM_PATH}/Tools/GNUTools/Make",
37+
"MSYS_path":"${config:MAXIM_PATH}/Tools/MSYS2",
3738

3839
"C_Cpp.default.includePath": [
3940
"${workspaceFolder}",
@@ -47,6 +48,7 @@
4748
"${config:MAXIM_PATH}/Libraries/PeriphDrivers/Include/${config:target}",
4849
"${config:MAXIM_PATH}/Libraries/MiscDrivers/Camera",
4950
"${config:MAXIM_PATH}/Libraries/MiscDrivers/Display",
51+
"${config:MAXIM_PATH}/Libraries/MiscDrivers/Display/fonts",
5052
"${config:MAXIM_PATH}/Libraries/MiscDrivers/ExtMemory",
5153
"${config:MAXIM_PATH}/Libraries/MiscDrivers/LED",
5254
"${config:MAXIM_PATH}/Libraries/MiscDrivers/PMIC",
@@ -60,13 +62,17 @@
6062
"${config:MAXIM_PATH}/Libraries/PeriphDrivers/Source",
6163
"${config:MAXIM_PATH}/Libraries/MiscDrivers/Camera",
6264
"${config:MAXIM_PATH}/Libraries/MiscDrivers/Display",
65+
"${config:MAXIM_PATH}/Libraries/MiscDrivers/Display/fonts",
6366
"${config:MAXIM_PATH}/Libraries/MiscDrivers/LED",
6467
"${config:MAXIM_PATH}/Libraries/MiscDrivers/PMIC",
6568
"${config:MAXIM_PATH}/Libraries/MiscDrivers/PushButton",
6669
"${config:MAXIM_PATH}/Libraries/MiscDrivers/Touchscreen",
6770
"${config:MAXIM_PATH}/Libraries/MiscDrivers"
6871
],
6972
"C_Cpp.default.defines": [
70-
"${config:board}"
73+
74+
],
75+
"C_Cpp.default.forcedInclude": [
76+
"${workspaceFolder}/build/project_defines.h"
7177
]
7278
}

‎assets/vscode/template/.vscode/flash.gdb

+3-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
define flash_m4
22
set architecture armv7e-m
3+
set remotetimeout 10
34
target remote | openocd -c "gdb_port pipe;log_output flash.log" -s $arg0/scripts -f interface/$arg1 -f target/$arg2 -c "init; reset halt"
45
load
56
compare-sections
@@ -8,8 +9,9 @@ end
89

910
define flash_m4_run
1011
set architecture armv7e-m
12+
set remotetimeout 10
1113
target remote | openocd -c "gdb_port pipe;log_output flash.log" -s $arg0/scripts -f interface/$arg1 -f target/$arg2 -c "init; reset halt"
1214
load
1315
compare-sections
1416
monitor resume
15-
end
17+
end

‎assets/vscode/template/.vscode/launch.json

+32-3
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,34 @@
11
{
22
"configurations": [
3+
{
4+
"name": "Debug Arm (Cortex-debug)",
5+
"cwd":"${workspaceRoot}",
6+
"executable": "${workspaceFolder}/build/${config:program_file}",
7+
"loadFiles": ["${workspaceFolder}/build/${config:program_file}"],
8+
"symbolFiles": [{
9+
"file": "${workspaceFolder}/build/${config:symbol_file}"
10+
}],
11+
"request": "launch",
12+
"type": "cortex-debug",
13+
"servertype": "openocd",
14+
"linux": {
15+
"gdbPath": "${config:ARM_GCC_path}/bin/arm-none-eabi-gdb",
16+
"serverpath": "${config:OCD_path}/openocd"
17+
},
18+
"windows": {
19+
"gdbPath": "${config:ARM_GCC_path}/bin/arm-none-eabi-gdb.exe",
20+
"serverpath": "${config:OCD_path}/openocd.exe"
21+
},
22+
"osx": {
23+
"gdbPath": "${config:ARM_GCC_path}/bin/arm-none-eabi-gdb",
24+
"serverpath": "${config:OCD_path}/openocd"
25+
},
26+
"searchDir": ["${config:OCD_path}/scripts"],
27+
"configFiles": ["interface/${config:M4_OCD_interface_file}", "target/${config:M4_OCD_target_file}"],
28+
"interface": "swd",
29+
"runToEntryPoint": "main",
30+
"svdFile": "${config:MAXIM_PATH}/Libraries/CMSIS/Device/Maxim/${config:target}/Include/${config:target}.svd"
31+
},
332
{
433
"name": "GDB (Arm M4)",
534
"type": "cppdbg",
@@ -62,15 +91,15 @@
6291
"externalConsole": false,
6392
"MIMode": "gdb",
6493
"linux": {
65-
"miDebuggerPath": "${config:xPack_GCC_path}/bin/riscv-none-embed-gdb",
94+
"miDebuggerPath": "${config:xPack_GCC_path}/bin/riscv-none-elf-gdb",
6695
"debugServerPath": "${config:OCD_path}/openocd"
6796
},
6897
"windows": {
69-
"miDebuggerPath": "${config:xPack_GCC_path}/bin/riscv-none-embed-gdb.exe",
98+
"miDebuggerPath": "${config:xPack_GCC_path}/bin/riscv-none-elf-gdb.exe",
7099
"debugServerPath": "${config:OCD_path}/openocd.exe"
71100
},
72101
"osx": {
73-
"miDebuggerPath": "${config:xPack_GCC_path}/bin/riscv-none-embed-gdb",
102+
"miDebuggerPath": "${config:xPack_GCC_path}/bin/riscv-none-elf-gdb",
74103
"debugServerPath": "${config:OCD_path}/bin/openocd"
75104
},
76105
"logging": {

‎assets/vscode/template/.vscode/tasks.json

+13-4
Original file line numberDiff line numberDiff line change
@@ -4,21 +4,30 @@
44
{
55
"label": "build",
66
"type": "shell",
7-
"command": "make -r -j 8 TARGET=${config:target} BOARD=${config:board} MAXIM_PATH=${config:MAXIM_PATH} MAKE=make PROJECT=${config:project_name}",
7+
"command": "make -r -j 8 --output-sync=target --no-print-directory TARGET=${config:target} BOARD=${config:board} MAXIM_PATH=${config:MAXIM_PATH} MAKE=make PROJECT=${config:project_name}",
8+
"osx":{
9+
"command": "source ~/.zshrc && make -r -j 8 --output-sync=target --no-print-directory TARGET=${config:target} BOARD=${config:board} MAXIM_PATH=${config:MAXIM_PATH} MAKE=make PROJECT=${config:project_name}"
10+
},
811
"group": "build",
912
"problemMatcher": []
1013
},
1114
{
1215
"label": "clean",
1316
"type": "shell",
14-
"command": "make -j 8 clean TARGET=${config:target} BOARD=${config:board} MAXIM_PATH=${config:MAXIM_PATH} MAKE=make PROJECT=${config:project_name}",
17+
"command": "make -j 8 clean --output-sync=target --no-print-directory TARGET=${config:target} BOARD=${config:board} MAXIM_PATH=${config:MAXIM_PATH} MAKE=make PROJECT=${config:project_name}",
18+
"osx":{
19+
"command": "source ~/.zshrc && make -j 8 clean --output-sync=target --no-print-directory TARGET=${config:target} BOARD=${config:board} MAXIM_PATH=${config:MAXIM_PATH} MAKE=make PROJECT=${config:project_name}"
20+
},
1521
"group": "build",
1622
"problemMatcher": []
1723
},
1824
{
1925
"label": "clean-periph",
2026
"type": "shell",
21-
"command": "make -j 8 distclean TARGET=${config:target} BOARD=${config:board} MAXIM_PATH=${config:MAXIM_PATH} MAKE=make PROJECT=${config:project_name}",
27+
"command": "make -j 8 distclean --output-sync=target --no-print-directory TARGET=${config:target} BOARD=${config:board} MAXIM_PATH=${config:MAXIM_PATH} MAKE=make PROJECT=${config:project_name}",
28+
"osx":{
29+
"command": "source ~/.zshrc && make -j 8 distclean --output-sync=target --no-print-directory TARGET=${config:target} BOARD=${config:board} MAXIM_PATH=${config:MAXIM_PATH} MAKE=make PROJECT=${config:project_name}"
30+
},
2231
"group": "build",
2332
"problemMatcher": []
2433
},
@@ -103,4 +112,4 @@
103112
"dependsOn":[]
104113
}
105114
]
106-
}
115+
}

‎assets/vscode/template/.vscode/templatesettings.json

+8-4
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,16 @@
11
{
22
"terminal.integrated.env.windows": {
3-
"Path":"${config:OCD_path};${config:ARM_GCC_path}/bin;${config:xPack_GCC_path}/bin;${config:Make_path};${env:PATH}",
3+
"Path":"${config:OCD_path};${config:ARM_GCC_path}/bin;${config:xPack_GCC_path}/bin;${config:MSYS_path}/usr/bin;${config:Make_path};${env:PATH}",
44
"MAXIM_PATH":"${config:MAXIM_PATH}"
55
},
66
"terminal.integrated.defaultProfile.windows": "Command Prompt",
7-
7+
88
"terminal.integrated.env.linux": {
9-
"PATH":"${config:OCD_path}:${config:ARM_GCC_path}/bin:${config:xPack_GCC_path}/bin:${env:PATH}",
9+
"PATH":"${config:OCD_path}:${config:ARM_GCC_path}/bin:${config:xPack_GCC_path}/bin:${config:Make_path}:${env:PATH}",
1010
"MAXIM_PATH":"${config:MAXIM_PATH}"
1111
},
1212
"terminal.integrated.env.osx": {
13-
"PATH":"${config:OCD_path}/bin:${config:ARM_GCC_path}/bin:${config:xPack_GCC_path}/bin:${env:PATH}",
13+
"PATH":"${config:OCD_path}/bin:${config:ARM_GCC_path}/bin:${config:xPack_GCC_path}/bin:${config:Make_path}:${env:PATH}",
1414
"MAXIM_PATH":"${config:MAXIM_PATH}"
1515
},
1616

@@ -34,6 +34,7 @@
3434
"ARM_GCC_path":"##__ARM_GCC_PATH__##",
3535
"xPack_GCC_path":"##__XPACK_GCC_PATH__##",
3636
"Make_path":"##__MAKE_PATH__##",
37+
"MSYS_path":"##__MSYS_PATH__##",
3738

3839
"C_Cpp.default.includePath": [
3940
"##__I_PATHS__##"
@@ -43,6 +44,9 @@
4344
],
4445
"C_Cpp.default.defines": [
4546
"##__DEFINES__##"
47+
],
48+
"C_Cpp.default.forcedInclude": [
49+
"${workspaceFolder}/build/project_defines.h"
4650
]
4751
}
4852

‎docs/shap.png

-182 KB
Binary file not shown.

‎docs/unload_example.png

64.8 KB
Loading

‎gen-demos-max78000.sh

+4-1
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,12 @@ python ai8xize.py --test-dir $TARGET --prefix cifar-100-simplewide2x-mixed --che
1212
python ai8xize.py --test-dir $TARGET --prefix cifar-100-residual --checkpoint-file trained/ai85-cifar100-residual-qat8-q.pth.tar --config-file networks/cifar100-ressimplenet.yaml --softmax $COMMON_ARGS --boost 2.5 "$@"
1313
python ai8xize.py --test-dir $TARGET --prefix kws20_v3 --checkpoint-file trained/ai85-kws20_v3-qat8-q.pth.tar --config-file networks/kws20-v3-hwc.yaml --softmax $COMMON_ARGS "$@"
1414
python ai8xize.py --test-dir $TARGET --prefix kws20_nas --checkpoint-file trained/ai85-kws20_nas-qat8-q.pth.tar --config-file networks/kws20-nas-hwc.yaml --softmax $COMMON_ARGS "$@"
15-
python ai8xize.py --test-dir $TARGET --prefix faceid --checkpoint-file trained/ai85-faceid-qat8-q.pth.tar --config-file networks/faceid.yaml --fifo $COMMON_ARGS "$@"
15+
python izer/add_fake_passthrough.py --input-checkpoint-path trained/ai85-faceid_112-qat-q.pth.tar --output-checkpoint-path trained/ai85-fakepass-faceid_112-qat-q.pth.tar --layer-name fakepass --layer-depth 128 --layer-name-after-pt linear --low-memory-footprint "$@"
16+
python ai8xize.py --test-dir $TARGET --prefix faceid_112 --checkpoint-file trained/ai85-fakepass-faceid_112-qat-q.pth.tar --config-file networks/ai85-faceid_112.yaml --fifo $COMMON_ARGS "$@"
1617
python ai8xize.py --test-dir $TARGET --prefix cats-dogs --checkpoint-file trained/ai85-catsdogs-qat8-q.pth.tar --config-file networks/cats-dogs-hwc.yaml --fifo --softmax $COMMON_ARGS "$@"
18+
python izer/add_fake_passthrough.py --input-checkpoint-path trained/ai85-camvid-unet-large-q.pth.tar --output-checkpoint-path trained/ai85-camvid-unet-large-fakept-q.pth.tar --layer-name pt --layer-depth 56 --layer-name-after-pt upconv3 "$@"
1719
python ai8xize.py --test-dir $TARGET --prefix camvid_unet --checkpoint-file trained/ai85-camvid-unet-large-fakept-q.pth.tar --config-file networks/camvid-unet-large-fakept.yaml $COMMON_ARGS --overlap-data --mlator --no-unload --max-checklines 8192 --new-kernel-loader "$@"
20+
python izer/add_fake_passthrough.py --input-checkpoint-path trained/ai85-aisegment-unet-large-q.pth.tar --output-checkpoint-path trained/ai85-aisegment-unet-large-fakept-q.pth.tar --layer-name pt --layer-depth 56 --layer-name-after-pt upconv3 "$@"
1821
python ai8xize.py --test-dir $TARGET --prefix aisegment_unet --checkpoint-file trained/ai85-aisegment-unet-large-fakept-q.pth.tar --config-file networks/aisegment-unet-large-fakept.yaml $COMMON_ARGS --overlap-data --mlator --no-unload --max-checklines 8192 --new-kernel-loader "$@"
1922
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 "$@"
2023
python ai8xize.py --test-dir $TARGET --prefix facedet_tinierssd --checkpoint-file trained/ai85-facedet-tinierssd-qat8-q.pth.tar --config-file networks/ai85-facedet-tinierssd.yaml --sample-input tests/sample_vggface2_facedetection.npy --fifo $COMMON_ARGS "$@"

‎gen-demos-max78002.sh

+5-3
Original file line numberDiff line numberDiff line change
@@ -12,14 +12,16 @@ python ai8xize.py --test-dir $TARGET --prefix cifar-100-simplewide2x-mixed --che
1212
python ai8xize.py --test-dir $TARGET --prefix cifar-100-residual --checkpoint-file trained/ai85-cifar100-residual-qat8-q.pth.tar --config-file networks/cifar100-ressimplenet.yaml --softmax $COMMON_ARGS "$@"
1313
python ai8xize.py --test-dir $TARGET --prefix kws20_v3_1 --checkpoint-file trained/ai87-kws20_v3-qat8-q.pth.tar --config-file networks/ai87-kws20-v3-hwc.yaml --softmax $COMMON_ARGS "$@"
1414
python ai8xize.py --test-dir $TARGET --prefix kws20_v2_1 --checkpoint-file trained/ai87-kws20_v2-qat8-q.pth.tar --config-file networks/ai87-kws20-v2-hwc.yaml --softmax $COMMON_ARGS "$@"
15-
python ai8xize.py --test-dir $TARGET --prefix faceid --checkpoint-file trained/ai85-faceid-qat8-q.pth.tar --config-file networks/faceid.yaml --fifo $COMMON_ARGS "$@"
15+
python ai8xize.py --test-dir $TARGET --prefix mobilefacenet-112 --checkpoint-file trained/ai87-mobilefacenet-112-qat-q.pth.tar --config-file networks/ai87-mobilefacenet-112.yaml --fifo $COMMON_ARGS "$@"
1616
python ai8xize.py --test-dir $TARGET --prefix cats-dogs --checkpoint-file trained/ai85-catsdogs-qat8-q.pth.tar --config-file networks/cats-dogs-hwc-no-fifo.yaml --softmax $COMMON_ARGS "$@"
17+
python izer/add_fake_passthrough.py --input-checkpoint-path trained/ai85-camvid-unet-large-q.pth.tar --output-checkpoint-path trained/ai85-camvid-unet-large-fakept-q.pth.tar --layer-name pt --layer-depth 56 --layer-name-after-pt upconv3 "$@"
1718
python ai8xize.py --test-dir $TARGET --prefix camvid_unet --checkpoint-file trained/ai85-camvid-unet-large-fakept-q.pth.tar --config-file networks/camvid-unet-large-fakept.yaml $COMMON_ARGS --overlap-data --mlator --no-unload --max-checklines 8192 "$@"
19+
python izer/add_fake_passthrough.py --input-checkpoint-path trained/ai85-aisegment-unet-large-q.pth.tar --output-checkpoint-path trained/ai85-aisegment-unet-large-fakept-q.pth.tar --layer-name pt --layer-depth 56 --layer-name-after-pt upconv3 "$@"
1820
python ai8xize.py --test-dir $TARGET --prefix aisegment_unet --checkpoint-file trained/ai85-aisegment-unet-large-fakept-q.pth.tar --config-file networks/aisegment-unet-large-fakept.yaml $COMMON_ARGS --overlap-data --mlator --no-unload --max-checklines 8192 "$@"
1921
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 "$@"
2022
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 "$@"
2123
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 "$@"
22-
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 "$@"
24+
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 "$@"
2325
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 "$@"
24-
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 --overwrite --no-unload $COMMON_ARGS "$@"
26+
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 "$@"
2527
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 "$@"

0 commit comments

Comments
 (0)