Add shelfnet. Resnet18backbone works

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-19 18:55:42 +02:00
parent 1dfc69ba89
commit 9f1e30eaa9
4 changed files with 219 additions and 1 deletions
@@ -102,6 +102,7 @@ int checkResult(int size, dnnType *data_d, dnnType *correct_d, bool device, int
    }
    int diffs = 0;
    for(int i=0; i<size; i++) {
+        // data_h[i] = data_h[i]*1e-2;
        if(data_h[i] != data_h[i] || correct_h[i] != correct_h[i] || //nan control
           fabs(data_h[i] - correct_h[i]) > eps) {
            diffs += 1;