Partial fixed

build/darknet/x64/partial.cmd

@@ -6,6 +6,9 @@ rem Download Yolo9000: http://pjreddie.com/media/files/yolo9000.weights
 rem darknet.exe partial cfg/tiny-yolo-voc.cfg tiny-yolo-voc.weights tiny-yolo-voc.conv.13 13
 
 
+darknet.exe partial cfg/darknet53_448.cfg darknet53_448.weights darknet53.conv.74 74
+
+
 darknet.exe partial cfg/yolov2-tiny-voc.cfg yolov2-tiny-voc.weights yolov2-tiny-voc.conv.13 13
 
 

src/classifier.c

@@ -110,7 +110,7 @@ void train_classifier(char *datacfg, char *cfgfile, char *weightfile, int *gpus,
     args.type = CLASSIFICATION_DATA;
 
 #ifdef OPENCV
-    args.threads = 3;
+    //args.threads = 3;
     IplImage* img = NULL;
     float max_img_loss = 5;
     int number_of_lines = 100;

src/parser.c

@@ -671,10 +671,13 @@ void parse_net_options(list *options, network *net)
     net->policy = get_policy(policy_s);
     net->burn_in = option_find_int_quiet(options, "burn_in", 0);
 #ifdef CUDNN_HALF
-    int compute_capability = get_gpu_compute_capability(net->gpu_index);
+    if (net->gpu_index >= 0) {
-    if(get_gpu_compute_capability(net->gpu_index) >= 700) net->cudnn_half = 1;
+        int compute_capability = get_gpu_compute_capability(net->gpu_index);
-    else net->cudnn_half = 0;
+        if (get_gpu_compute_capability(net->gpu_index) >= 700) net->cudnn_half = 1;
-    fprintf(stderr, " compute_capability = %d, cudnn_half = %d \n", compute_capability, net->cudnn_half);
+        else net->cudnn_half = 0;
+        fprintf(stderr, " compute_capability = %d, cudnn_half = %d \n", compute_capability, net->cudnn_half);
+    }
+    else fprintf(stderr, " GPU isn't used \n");
 #endif
     if(net->policy == STEP){
         net->step = option_find_int(options, "step", 1);