batch seems ok in yolo3_berkely

layers to be checked: DeformableConvRT FlattenConcatRT ReshapeRT RouteRT (dont know why but seems working)
2020-04-21 19:41:40 +00:00
parent ef91439c87
commit a91b2f46ba
10 changed files with 51 additions and 21 deletions
@@ -41,7 +41,7 @@ public:
 	virtual int enqueue(int batchSize, const void*const * inputs, void** outputs, void* workspace, cudaStream_t stream) override {

 		activationReLUCeilingForward((dnnType*)reinterpret_cast<const dnnType*>(inputs[0]), 
-											reinterpret_cast<dnnType*>(outputs[0]), size, ceiling, stream);
+											reinterpret_cast<dnnType*>(outputs[0]), batchSize*size, ceiling, stream);
 		return 0;
 	}