conv2d ok, but deconv ha different dim with tensorrt

This commit is contained in:
Francesco Gatti
2019-10-30 16:45:46 +01:00
parent f247300469
commit 2594f59d0d
4 changed files with 9 additions and 8 deletions
+2
View File
@@ -19,6 +19,8 @@ int main() {
tk::dnn::Dense l5(&net, 4, d2_bin);
tk::dnn::Activation l6(&net, CUDNN_ACTIVATION_RELU);
net.print();
// Load input
dnnType *data;
dnnType *input_h;