Minor fix

This commit is contained in:
AlexeyAB
2018-10-18 00:04:26 +03:00
parent 76706dc4a6
commit 3b59178dec

View File

@ -108,7 +108,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
args.small_object = net.small_object; args.small_object = net.small_object;
args.d = &buffer; args.d = &buffer;
args.type = DETECTION_DATA; args.type = DETECTION_DATA;
args.threads = 16; // 64 args.threads = 64; // 16 or 64
args.angle = net.angle; args.angle = net.angle;
args.exposure = net.exposure; args.exposure = net.exposure;
@ -116,7 +116,8 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
args.hue = net.hue; args.hue = net.hue;
#ifdef OPENCV #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; IplImage* img = NULL;
float max_img_loss = 5; float max_img_loss = 5;
int number_of_lines = 100; int number_of_lines = 100;