batch size > 1

2020-04-21 19:20:44 +02:00
parent ecb2f5e3ad
commit ef91439c87
6 changed files with 51 additions and 22 deletions
@@ -62,6 +62,7 @@ public:
    dataDim_t getOutputDim();

    bool fp16, dla, int8;
+    int maxBatchSize;
    bool dontLoadWeights;
    std::string fileImgList;
    std::string fileLabelList;