RegionRT softmax fix

2020-04-24 00:08:06 +02:00
parent 1ea6097a97
commit 13c9dc6620
1 changed files with 1 additions and 1 deletions
@@ -61,7 +61,7 @@ public:
 		//softmax start
 		int index = entry_index(0, 0, coords + 1);
 		softmaxForward(	srcData + index, classes, batchSize*num, 
-						(batchSize*c*h*w)/num, 
+						(c*h*w)/num, 
 						w*h, 1, w*h, 1, dstData + index, stream);

 		return 0;