Tino Hager
2018-06-27 8a504c737d2bc3b9b37cb79cb50fbf7eecda07df
src/detector.c
@@ -61,10 +61,14 @@
    srand(time(0));
    network net = nets[0];
   if ((net.batch * net.subdivisions) == 1) {
   const int actual_batch_size = net.batch * net.subdivisions;
   if (actual_batch_size == 1) {
      printf("\n Error: You set incorrect value batch=1 for Training! You should set batch=64 subdivision=64 \n");
      getchar();
   }
   else if (actual_batch_size < 64) {
         printf("\n Warning: You set batch=%d lower than 64! It is recommended to set batch=64 subdivision=64 \n", actual_batch_size);
   }
    int imgs = net.batch * net.subdivisions * ngpus;
    printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
@@ -1092,7 +1096,7 @@
   fuse_conv_batchnorm(net);
   if (net.layers[net.n - 1].classes != names_size) {
      printf(" Error: in the file %s number of names %d that isn't equal to classes=%d in the file %s \n", 
         name_list, names_size, net.layers[net.n - 1].classes, datacfg);
         name_list, names_size, net.layers[net.n - 1].classes, cfgfile);
      if(net.layers[net.n - 1].classes > names_size) getchar();
   }
    srand(2222222);