mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
238 Commits 11 Branches 5 Tags
f6129311d62b72a751d0872c4e8ad09ffc4d7f4a
Commit Graph

2 Commits

Author SHA1 Message Date
Micaela Verucchi 1a50065bde Fix conv2d with additional bias for tensorRT. Fix reshape deserialize. Mb2512 works with tensorRT 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-17 13:18:05 +01:00
Micaela Verucchi 14d850751d Add mobilenetv2ssdlite512 test. 
Works for CuDNN, not for tensorRT.
Modified channels in second convolution for classification headers from 126 to 486, when changing size from 300 to 512.
Added size 512 SSD specs and support to COCO dataset (81 classes, first BACKGROUND due to repo for training).
Refactoring class names.

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-03-17 10:35:02 +01:00