Fixed resizing REORG_OLD layer

pull/3270/head
AlexeyAB 6 years ago
parent 2b7fea761f
commit 24f7d94ab4
  1. 2
      src/detector.c
  2. 2
      src/network.c

@ -86,7 +86,7 @@ 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 (actual_batch_size < 64) {
else if (actual_batch_size < 8) {
printf("\n Warning: You set batch=%d lower than 64! It is recommended to set batch=64 subdivision=64 \n", actual_batch_size);
}

@ -523,6 +523,8 @@ int resize_network(network *net, int w, int h)
resize_upsample_layer(&l, w, h);
}else if(l.type == REORG){
resize_reorg_layer(&l, w, h);
} else if (l.type == REORG_OLD) {
resize_reorg_old_layer(&l, w, h);
}else if(l.type == AVGPOOL){
resize_avgpool_layer(&l, w, h);
}else if(l.type == NORMALIZATION){

Loading…
Cancel
Save