Minor fix

This commit is contained in:
AlexeyAB
2018-06-13 13:50:02 +03:00
parent eff487ba36
commit 6d44529cf9
3 changed files with 10 additions and 8 deletions

View File

@ -65,6 +65,9 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
printf("\n Error: You set incorrect value batch=1 for Training! You should set batch=64 subdivision=64 \n");
getchar();
}
else if ((net.batch * net.subdivisions) < 64) {
printf("\n Warning: You set batch= lower than 64! It is recommended to set batch=64 subdivision=64 \n");
}
int imgs = net.batch * net.subdivisions * ngpus;
printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);