Micaela Verucchi
2fbac7705d
Move download of weights inside build folder
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-17 22:36:25 +02:00
Davide Sapienza
106cbb5c73
Remove the optional parameters from the Pooling layer.
...
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-04-09 19:20:27 +02:00
Micaela Verucchi
fabad7bace
Add mnist download, change ERROR_TKDNN into ERROR_TENSORRT
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-09 19:02:25 +02:00
Micaela Verucchi
3502c5b676
Fix pooling, add return code in each test, add return code handling in test_all_tests script
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-09 15:25:05 +02:00
Davide Sapienza
ce20868bad
Add TKDNN_MODE variable to the name of the network.
...
This commit permits to obtain different .rt files for different
precision optimizations of the same network.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com >
2020-03-30 15:08:25 +02:00
Micaela Verucchi
7c55dcb708
Add tp tests for yolo3512 and yolo3tiny512
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-03-26 15:06:49 +01:00