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

adding reorg3d layer

Browse Source
This commit is contained in:
MohammadReza Hosseini
2020-06-14 18:48:25 +04:30
parent b8ecd3c0cb
commit e20d3a0861
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/tkDNN/DarknetParser.h
+1 -1
View File
@@ -182,7 +182,7 @@ namespace tk { namespace dnn {
} }
netLayers.push_back(new tk::dnn::Route(net, layers.data(), layers.size())); netLayers.push_back(new tk::dnn::Route(net, layers.data(), layers.size()));
} else if(f.type == "reorg") { } else if(f.type == "reorg" || f.type == "reorg3d") {
netLayers.push_back(new tk::dnn::Reorg(net, f.stride_x)); netLayers.push_back(new tk::dnn::Reorg(net, f.stride_x));
} else if(f.type == "region") { } else if(f.type == "region") {