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

Adding dilated convolution and support for reorg3d layer #47

Open
mrhosseini wants to merge 8 commits from mrhosseini/master into master
pull from: mrhosseini/master
merge into: mmr:master
Conversation 1 Commits 8 Files Changed 8
+1 -1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit e20d3a0861 - Show all commits
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()));
} else if(f.type == "reorg") {
} else if(f.type == "reorg" || f.type == "reorg3d") {
netLayers.push_back(new tk::dnn::Reorg(net, f.stride_x));
} else if(f.type == "region") {