mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

RegionRT softmax fix

Browse Source
This commit is contained in:
Francesco Gatti
2020-04-24 00:08:06 +02:00
parent afaeb75d84
commit 6a185564e7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/tkDNN/pluginsRT/RegionRT.h
+1 -1
View File
@@ -61,7 +61,7 @@ public:
//softmax start //softmax start
int index = entry_index(0, 0, coords + 1); int index = entry_index(0, 0, coords + 1);
softmaxForward( srcData + index, classes, batchSize*num, softmaxForward( srcData + index, classes, batchSize*num,
(batchSize*c*h*w)/num, (c*h*w)/num,
w*h, 1, w*h, 1, dstData + index, stream); w*h, 1, w*h, 1, dstData + index, stream);
return 0; return 0;