Fix for multi-GPU and DGX

pull/981/head^2
AlexeyAB 7 years ago
parent 4cf4a89f05
commit 483fdfa951
  1. 2
      src/detector.c

@ -106,7 +106,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
args.hue = net.hue;
#ifdef OPENCV
args.threads = 3;
args.threads = 3 * ngpus;
IplImage* img = NULL;
float max_img_loss = 5;
int number_of_lines = 100;

Loading…
Cancel
Save