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

yolo3 layer 15

Browse Source
This commit is contained in:
Francesco Gatti
2018-12-20 11:36:10 +01:00
parent e91db28756
commit 7a46601306
3 changed files with 42 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Shortcut.cpp
+4 -2
View File
@@ -5,7 +5,7 @@
namespace tk { namespace dnn {
Shortcut::Shortcut(Network *net, Layer *backLayer, int layers_n) : Layer(net) {
Shortcut::Shortcut(Network *net, Layer *backLayer) : Layer(net) {
this->backLayer = backLayer;
checkCuda( cudaMalloc(&dstData, output_dim.tot()*sizeof(dnnType)) );
@@ -18,8 +18,10 @@ Shortcut::~Shortcut() {
dnnType* Shortcut::infer(dataDim_t &dim, dnnType* srcData) {
dataDim_t bdim = this->backLayer->output_dim;
checkCuda(cudaMemcpy(dstData, srcData, dim.tot()*sizeof(dnnType), cudaMemcpyDeviceToDevice));
shortcutForward(this->backLayer->dstData, dstData, dim.n, dim.c, dim.h, dim.w, 1, bdim.n, bdim.c, bdim.h, bdim.w, 1);
//update data dimensions
dim = output_dim;