From 351996d9f0390ef44412fa678bc7a073a94e23e5 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Thu, 16 Mar 2017 18:49:36 +0000 Subject: [PATCH] Fixed memory leak in DLL, added load_image() & free_image(), added read_names_from_file() --- src/convolutional_layer.c | 4 ++++ 1 files changed, 4 insertions(+), 0 deletions(-) diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c index 37211ab..cf5d252 100644 --- a/src/convolutional_layer.c +++ b/src/convolutional_layer.c @@ -8,6 +8,10 @@ #include <stdio.h> #include <time.h> +#ifdef CUDNN +#pragma comment(lib, "cudnn.lib") +#endif + #ifdef AI2 #include "xnor_layer.h" #endif -- Gitblit v1.10.0