diff --git a/src/detector.c b/src/detector.c index c750ff7a..bf8d1181 100644 --- a/src/detector.c +++ b/src/detector.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); } diff --git a/src/network.c b/src/network.c index 86c248c4..d54727b3 100644 --- a/src/network.c +++ b/src/network.c @@ -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){