From b9ea49af250a3eab3b8775efa53db0f0ff063357 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Mon, 20 May 2019 23:58:55 +0300 Subject: [PATCH] Fixed reorg layer is [reorg3d], old reorg layer is [reorg] --- src/parser.c | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/parser.c b/src/parser.c index ac7fffb7..56a0bcb7 100644 --- a/src/parser.c +++ b/src/parser.c @@ -68,8 +68,8 @@ LAYER_TYPE string_to_layer_type(char * type) || strcmp(type, "[connected]")==0) return CONNECTED; if (strcmp(type, "[max]")==0 || strcmp(type, "[maxpool]")==0) return MAXPOOL; - if (strcmp(type, "[reorg]")==0) return REORG; - if (strcmp(type, "[reorg_old]") == 0) return REORG_OLD; + if (strcmp(type, "[reorg3d]")==0) return REORG; + if (strcmp(type, "[reorg]") == 0) return REORG_OLD; if (strcmp(type, "[avg]")==0 || strcmp(type, "[avgpool]")==0) return AVGPOOL; if (strcmp(type, "[dropout]")==0) return DROPOUT;