Softmax

2017-07-18 16:45:19 +02:00
parent 0887199880
commit e7a6f1fb6c
4 changed files with 69 additions and 1 deletions
@@ -35,6 +35,8 @@ Activation::Activation(Network *net, dataDim_t input_dim, cudnnActivationMode_t
 Activation::~Activation() {

    checkCuda( cudaFree(dstData) );
+
+    checkCUDNN( cudnnDestroyActivationDescriptor(activDesc) );
 }

 value_type* Activation::infer(dataDim_t &dim, value_type* srcData) {