From 5cfca657b966998d164aecf1d0698ff69e783b18 Mon Sep 17 00:00:00 2001 From: Fede Paredes Date: Thu, 10 Jan 2019 13:50:37 +0100 Subject: [PATCH] summary() fixed --- src/cusnn.cu | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/cusnn.cu b/src/cusnn.cu index 8b2efaf..9d94b88 100644 --- a/src/cusnn.cu +++ b/src/cusnn.cu @@ -1064,7 +1064,8 @@ void Network::summary() { printf("Input \t\t\t (%i, %i, %i)\n", this->h_inp_size[0], this->h_inp_size[1], this->h_inp_size[2]); printf("----------------------------------------------------------\n"); for (int l = 0; l < this->cnt_layers; l++) { - int params = (this->h_layers[l]->rf_side - this->h_layers[l]->rf_side_limits[0]) * + int params = this->h_layers[l]->inp_size[0] * + (this->h_layers[l]->rf_side - this->h_layers[l]->rf_side_limits[0]) * (this->h_layers[l]->rf_side - this->h_layers[l]->rf_side_limits[1]) * this->h_layers[l]->num_delays; for (int k = 0; k < this->h_layers[l]->cnt_kernels; k++) { if (!this->h_layers[l]->h_kernels_cnvg[k] && this->h_layers[l]->enable_learning) trainable += params;