Loss-graph store automatically (iterations == max_batches) at the end of training

This commit is contained in:
AlexeyAB
2018-04-16 13:09:10 +03:00
parent 6aa140b3c6
commit c52fa47428
3 changed files with 3 additions and 3 deletions

View File

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