batch size > 1

2020-04-21 19:20:44 +02:00
parent b1818b81d9
commit 7c81c5a43c
6 changed files with 51 additions and 22 deletions
@@ -34,6 +34,10 @@ Network::Network(dataDim_t input_dim) {
            int8 = true;
        }
    }
+    maxBatchSize = 1;
+    if(const char* env_p = std::getenv("TKDNN_BATCHSIZE")) {
+        maxBatchSize = atoi(env_p);
+    }
    if(const char* env_p = std::getenv("TKDNN_CALIB_IMG_PATH"))
        fileImgList = env_p;