From 3b59178dec5e9b105b2fba23001c768206da0276 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Thu, 18 Oct 2018 00:04:26 +0300 Subject: [PATCH] Minor fix --- src/detector.c | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/detector.c b/src/detector.c index 5926979f..fc560f80 100644 --- a/src/detector.c +++ b/src/detector.c @@ -108,7 +108,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i args.small_object = net.small_object; args.d = &buffer; args.type = DETECTION_DATA; - args.threads = 16; // 64 + args.threads = 64; // 16 or 64 args.angle = net.angle; args.exposure = net.exposure; @@ -116,7 +116,8 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i args.hue = net.hue; #ifdef OPENCV - args.threads = 3 * ngpus; + args.threads = 3 * ngpus; // Amazon EC2 Tesla V100: p3.2xlarge (8 logical cores) - p3.16xlarge + //args.threads = 12 * ngpus; // Ryzen 7 2700X (16 logical cores) IplImage* img = NULL; float max_img_loss = 5; int number_of_lines = 100;