mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Move download of weights inside build folder

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-04-17 22:36:25 +02:00
parent f637aa0ea8
commit 2fbac7705d
30 changed files with 820 additions and 818 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/mnist/test_mnistRT.cpp
+7 -7
View File
@@ -3,12 +3,12 @@
#include "tkdnn.h"
#include "NvInfer.h"
const char *input_bin = "../tests/mnist/input.bin";
const char *c0_bin = "../tests/mnist/layers/c0.bin";
const char *c1_bin = "../tests/mnist/layers/c1.bin";
const char *d2_bin = "../tests/mnist/layers/d2.bin";
const char *d3_bin = "../tests/mnist/layers/d3.bin";
const char *output_bin = "../tests/mnist/output.bin";
const char *input_bin = "mnist/input.bin";
const char *c0_bin = "mnist/layers/c0.bin";
const char *c1_bin = "mnist/layers/c1.bin";
const char *d2_bin = "mnist/layers/d2.bin";
const char *d3_bin = "mnist/layers/d3.bin";
const char *output_bin = "mnist/output.bin";
using namespace nvinfer1;
@@ -25,7 +25,7 @@ class Logger : public ILogger
int main() {
downloadWeightsifDoNotExist(input_bin, "../tests/mnist", "https://cloud.hipert.unimore.it/s/2TyQkMJL3LArLAS/download");
downloadWeightsifDoNotExist(input_bin, "mnist", "https://cloud.hipert.unimore.it/s/2TyQkMJL3LArLAS/download");
std::cout<<"\n==== CUDNN ====\n";
// Network layout