Add mnist download, change ERROR_TKDNN into ERROR_TENSORRT

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-04-09 19:02:25 +02:00
parent 8a575f6447
commit a8fd4cabf3
28 changed files with 36 additions and 32 deletions
+2
View File
@@ -25,6 +25,8 @@ class Logger : public ILogger
int main() {
downloadWeightsifDoNotExist(input_bin, "../tests/mnist", "https://cloud.hipert.unimore.it/s/2TyQkMJL3LArLAS/download");
std::cout<<"\n==== CUDNN ====\n";
// Network layout
tk::dnn::dataDim_t dim(1, 1, 28, 28, 1);