diff --git a/CNNDemo/ResNet50Weight.bin b/CNNDemo/ResNet50Weight.bin
index f267736..15dbddd 100644
Binary files a/CNNDemo/ResNet50Weight.bin and b/CNNDemo/ResNet50Weight.bin differ
diff --git a/CNNDemo/ResNet50Weight.cc b/CNNDemo/ResNet50Weight.cc
index 04e17f5..b9cca78 100644
--- a/CNNDemo/ResNet50Weight.cc
+++ b/CNNDemo/ResNet50Weight.cc
@@ -413,211 +413,3 @@ int verify_layer3[] = { 124357344,125160159 };
 int verify_layer4[] = { 125160160,125561567 };
 int verify_avgpool[] = { 125561568,125569759 };
 int verify_fc[] = { 125569760,125573759 };
-int layer1_block0_conv1_input[] = { 125573760,126376575 };
-int layer1_block0_conv1_output[] = { 126376576,127179391 };
-int layer1_block0_bn1_input[] = { 127179392,127982207 };
-int layer1_block0_bn1_output[] = { 127982208,128785023 };
-int layer1_block0_conv2_input[] = { 128785024,129587839 };
-int layer1_block0_conv2_output[] = { 129587840,130390655 };
-int layer1_block0_bn2_input[] = { 130390656,131193471 };
-int layer1_block0_bn2_output[] = { 131193472,131996287 };
-int layer1_block0_conv3_input[] = { 131996288,132799103 };
-int layer1_block0_conv3_output[] = { 132799104,136010367 };
-int layer1_block0_bn3_input[] = { 136010368,139221631 };
-int layer1_block0_bn3_output[] = { 139221632,142432895 };
-int layer1_block0_downsample_conv_input[] = { 142432896,143235711 };
-int layer1_block0_downsample_conv_output[] = { 143235712,146446975 };
-int layer1_block0_downsample_bn_input[] = { 146446976,149658239 };
-int layer1_block0_downsample_bn_output[] = { 149658240,152869503 };
-int layer1_block1_conv1_input[] = { 152869504,156080767 };
-int layer1_block1_conv1_output[] = { 156080768,156883583 };
-int layer1_block1_bn1_input[] = { 156883584,157686399 };
-int layer1_block1_bn1_output[] = { 157686400,158489215 };
-int layer1_block1_conv2_input[] = { 158489216,159292031 };
-int layer1_block1_conv2_output[] = { 159292032,160094847 };
-int layer1_block1_bn2_input[] = { 160094848,160897663 };
-int layer1_block1_bn2_output[] = { 160897664,161700479 };
-int layer1_block1_conv3_input[] = { 161700480,162503295 };
-int layer1_block1_conv3_output[] = { 162503296,165714559 };
-int layer1_block1_bn3_input[] = { 165714560,168925823 };
-int layer1_block1_bn3_output[] = { 168925824,172137087 };
-int layer1_block2_conv1_input[] = { 172137088,175348351 };
-int layer1_block2_conv1_output[] = { 175348352,176151167 };
-int layer1_block2_bn1_input[] = { 176151168,176953983 };
-int layer1_block2_bn1_output[] = { 176953984,177756799 };
-int layer1_block2_conv2_input[] = { 177756800,178559615 };
-int layer1_block2_conv2_output[] = { 178559616,179362431 };
-int layer1_block2_bn2_input[] = { 179362432,180165247 };
-int layer1_block2_bn2_output[] = { 180165248,180968063 };
-int layer1_block2_conv3_input[] = { 180968064,181770879 };
-int layer1_block2_conv3_output[] = { 181770880,184982143 };
-int layer1_block2_bn3_input[] = { 184982144,188193407 };
-int layer1_block2_bn3_output[] = { 188193408,191404671 };
-int layer2_block0_conv1_input[] = { 191404672,194615935 };
-int layer2_block0_conv1_output[] = { 194615936,196221567 };
-int layer2_block0_bn1_input[] = { 196221568,197827199 };
-int layer2_block0_bn1_output[] = { 197827200,199432831 };
-int layer2_block0_conv2_input[] = { 199432832,201038463 };
-int layer2_block0_conv2_output[] = { 201038464,201439871 };
-int layer2_block0_bn2_input[] = { 201439872,201841279 };
-int layer2_block0_bn2_output[] = { 201841280,202242687 };
-int layer2_block0_conv3_input[] = { 202242688,202644095 };
-int layer2_block0_conv3_output[] = { 202644096,204249727 };
-int layer2_block0_bn3_input[] = { 204249728,205855359 };
-int layer2_block0_bn3_output[] = { 205855360,207460991 };
-int layer2_block0_downsample_conv_input[] = { 207460992,210672255 };
-int layer2_block0_downsample_conv_output[] = { 210672256,212277887 };
-int layer2_block0_downsample_bn_input[] = { 212277888,213883519 };
-int layer2_block0_downsample_bn_output[] = { 213883520,215489151 };
-int layer2_block1_conv1_input[] = { 215489152,217094783 };
-int layer2_block1_conv1_output[] = { 217094784,217496191 };
-int layer2_block1_bn1_input[] = { 217496192,217897599 };
-int layer2_block1_bn1_output[] = { 217897600,218299007 };
-int layer2_block1_conv2_input[] = { 218299008,218700415 };
-int layer2_block1_conv2_output[] = { 218700416,219101823 };
-int layer2_block1_bn2_input[] = { 219101824,219503231 };
-int layer2_block1_bn2_output[] = { 219503232,219904639 };
-int layer2_block1_conv3_input[] = { 219904640,220306047 };
-int layer2_block1_conv3_output[] = { 220306048,221911679 };
-int layer2_block1_bn3_input[] = { 221911680,223517311 };
-int layer2_block1_bn3_output[] = { 223517312,225122943 };
-int layer2_block2_conv1_input[] = { 225122944,226728575 };
-int layer2_block2_conv1_output[] = { 226728576,227129983 };
-int layer2_block2_bn1_input[] = { 227129984,227531391 };
-int layer2_block2_bn1_output[] = { 227531392,227932799 };
-int layer2_block2_conv2_input[] = { 227932800,228334207 };
-int layer2_block2_conv2_output[] = { 228334208,228735615 };
-int layer2_block2_bn2_input[] = { 228735616,229137023 };
-int layer2_block2_bn2_output[] = { 229137024,229538431 };
-int layer2_block2_conv3_input[] = { 229538432,229939839 };
-int layer2_block2_conv3_output[] = { 229939840,231545471 };
-int layer2_block2_bn3_input[] = { 231545472,233151103 };
-int layer2_block2_bn3_output[] = { 233151104,234756735 };
-int layer2_block3_conv1_input[] = { 234756736,236362367 };
-int layer2_block3_conv1_output[] = { 236362368,236763775 };
-int layer2_block3_bn1_input[] = { 236763776,237165183 };
-int layer2_block3_bn1_output[] = { 237165184,237566591 };
-int layer2_block3_conv2_input[] = { 237566592,237967999 };
-int layer2_block3_conv2_output[] = { 237968000,238369407 };
-int layer2_block3_bn2_input[] = { 238369408,238770815 };
-int layer2_block3_bn2_output[] = { 238770816,239172223 };
-int layer2_block3_conv3_input[] = { 239172224,239573631 };
-int layer2_block3_conv3_output[] = { 239573632,241179263 };
-int layer2_block3_bn3_input[] = { 241179264,242784895 };
-int layer2_block3_bn3_output[] = { 242784896,244390527 };
-int layer3_block0_conv1_input[] = { 244390528,245996159 };
-int layer3_block0_conv1_output[] = { 245996160,246798975 };
-int layer3_block0_bn1_input[] = { 246798976,247601791 };
-int layer3_block0_bn1_output[] = { 247601792,248404607 };
-int layer3_block0_conv2_input[] = { 248404608,249207423 };
-int layer3_block0_conv2_output[] = { 249207424,249408127 };
-int layer3_block0_bn2_input[] = { 249408128,249608831 };
-int layer3_block0_bn2_output[] = { 249608832,249809535 };
-int layer3_block0_conv3_input[] = { 249809536,250010239 };
-int layer3_block0_conv3_output[] = { 250010240,250813055 };
-int layer3_block0_bn3_input[] = { 250813056,251615871 };
-int layer3_block0_bn3_output[] = { 251615872,252418687 };
-int layer3_block0_downsample_conv_input[] = { 252418688,254024319 };
-int layer3_block0_downsample_conv_output[] = { 254024320,254827135 };
-int layer3_block0_downsample_bn_input[] = { 254827136,255629951 };
-int layer3_block0_downsample_bn_output[] = { 255629952,256432767 };
-int layer3_block1_conv1_input[] = { 256432768,257235583 };
-int layer3_block1_conv1_output[] = { 257235584,257436287 };
-int layer3_block1_bn1_input[] = { 257436288,257636991 };
-int layer3_block1_bn1_output[] = { 257636992,257837695 };
-int layer3_block1_conv2_input[] = { 257837696,258038399 };
-int layer3_block1_conv2_output[] = { 258038400,258239103 };
-int layer3_block1_bn2_input[] = { 258239104,258439807 };
-int layer3_block1_bn2_output[] = { 258439808,258640511 };
-int layer3_block1_conv3_input[] = { 258640512,258841215 };
-int layer3_block1_conv3_output[] = { 258841216,259644031 };
-int layer3_block1_bn3_input[] = { 259644032,260446847 };
-int layer3_block1_bn3_output[] = { 260446848,261249663 };
-int layer3_block2_conv1_input[] = { 261249664,262052479 };
-int layer3_block2_conv1_output[] = { 262052480,262253183 };
-int layer3_block2_bn1_input[] = { 262253184,262453887 };
-int layer3_block2_bn1_output[] = { 262453888,262654591 };
-int layer3_block2_conv2_input[] = { 262654592,262855295 };
-int layer3_block2_conv2_output[] = { 262855296,263055999 };
-int layer3_block2_bn2_input[] = { 263056000,263256703 };
-int layer3_block2_bn2_output[] = { 263256704,263457407 };
-int layer3_block2_conv3_input[] = { 263457408,263658111 };
-int layer3_block2_conv3_output[] = { 263658112,264460927 };
-int layer3_block2_bn3_input[] = { 264460928,265263743 };
-int layer3_block2_bn3_output[] = { 265263744,266066559 };
-int layer3_block3_conv1_input[] = { 266066560,266869375 };
-int layer3_block3_conv1_output[] = { 266869376,267070079 };
-int layer3_block3_bn1_input[] = { 267070080,267270783 };
-int layer3_block3_bn1_output[] = { 267270784,267471487 };
-int layer3_block3_conv2_input[] = { 267471488,267672191 };
-int layer3_block3_conv2_output[] = { 267672192,267872895 };
-int layer3_block3_bn2_input[] = { 267872896,268073599 };
-int layer3_block3_bn2_output[] = { 268073600,268274303 };
-int layer3_block3_conv3_input[] = { 268274304,268475007 };
-int layer3_block3_conv3_output[] = { 268475008,269277823 };
-int layer3_block3_bn3_input[] = { 269277824,270080639 };
-int layer3_block3_bn3_output[] = { 270080640,270883455 };
-int layer3_block4_conv1_input[] = { 270883456,271686271 };
-int layer3_block4_conv1_output[] = { 271686272,271886975 };
-int layer3_block4_bn1_input[] = { 271886976,272087679 };
-int layer3_block4_bn1_output[] = { 272087680,272288383 };
-int layer3_block4_conv2_input[] = { 272288384,272489087 };
-int layer3_block4_conv2_output[] = { 272489088,272689791 };
-int layer3_block4_bn2_input[] = { 272689792,272890495 };
-int layer3_block4_bn2_output[] = { 272890496,273091199 };
-int layer3_block4_conv3_input[] = { 273091200,273291903 };
-int layer3_block4_conv3_output[] = { 273291904,274094719 };
-int layer3_block4_bn3_input[] = { 274094720,274897535 };
-int layer3_block4_bn3_output[] = { 274897536,275700351 };
-int layer3_block5_conv1_input[] = { 275700352,276503167 };
-int layer3_block5_conv1_output[] = { 276503168,276703871 };
-int layer3_block5_bn1_input[] = { 276703872,276904575 };
-int layer3_block5_bn1_output[] = { 276904576,277105279 };
-int layer3_block5_conv2_input[] = { 277105280,277305983 };
-int layer3_block5_conv2_output[] = { 277305984,277506687 };
-int layer3_block5_bn2_input[] = { 277506688,277707391 };
-int layer3_block5_bn2_output[] = { 277707392,277908095 };
-int layer3_block5_conv3_input[] = { 277908096,278108799 };
-int layer3_block5_conv3_output[] = { 278108800,278911615 };
-int layer3_block5_bn3_input[] = { 278911616,279714431 };
-int layer3_block5_bn3_output[] = { 279714432,280517247 };
-int layer4_block0_conv1_input[] = { 280517248,281320063 };
-int layer4_block0_conv1_output[] = { 281320064,281721471 };
-int layer4_block0_bn1_input[] = { 281721472,282122879 };
-int layer4_block0_bn1_output[] = { 282122880,282524287 };
-int layer4_block0_conv2_input[] = { 282524288,282925695 };
-int layer4_block0_conv2_output[] = { 282925696,283026047 };
-int layer4_block0_bn2_input[] = { 283026048,283126399 };
-int layer4_block0_bn2_output[] = { 283126400,283226751 };
-int layer4_block0_conv3_input[] = { 283226752,283327103 };
-int layer4_block0_conv3_output[] = { 283327104,283728511 };
-int layer4_block0_bn3_input[] = { 283728512,284129919 };
-int layer4_block0_bn3_output[] = { 284129920,284531327 };
-int layer4_block0_downsample_conv_input[] = { 284531328,285334143 };
-int layer4_block0_downsample_conv_output[] = { 285334144,285735551 };
-int layer4_block0_downsample_bn_input[] = { 285735552,286136959 };
-int layer4_block0_downsample_bn_output[] = { 286136960,286538367 };
-int layer4_block1_conv1_input[] = { 286538368,286939775 };
-int layer4_block1_conv1_output[] = { 286939776,287040127 };
-int layer4_block1_bn1_input[] = { 287040128,287140479 };
-int layer4_block1_bn1_output[] = { 287140480,287240831 };
-int layer4_block1_conv2_input[] = { 287240832,287341183 };
-int layer4_block1_conv2_output[] = { 287341184,287441535 };
-int layer4_block1_bn2_input[] = { 287441536,287541887 };
-int layer4_block1_bn2_output[] = { 287541888,287642239 };
-int layer4_block1_conv3_input[] = { 287642240,287742591 };
-int layer4_block1_conv3_output[] = { 287742592,288143999 };
-int layer4_block1_bn3_input[] = { 288144000,288545407 };
-int layer4_block1_bn3_output[] = { 288545408,288946815 };
-int layer4_block2_conv1_input[] = { 288946816,289348223 };
-int layer4_block2_conv1_output[] = { 289348224,289448575 };
-int layer4_block2_bn1_input[] = { 289448576,289548927 };
-int layer4_block2_bn1_output[] = { 289548928,289649279 };
-int layer4_block2_conv2_input[] = { 289649280,289749631 };
-int layer4_block2_conv2_output[] = { 289749632,289849983 };
-int layer4_block2_bn2_input[] = { 289849984,289950335 };
-int layer4_block2_bn2_output[] = { 289950336,290050687 };
-int layer4_block2_conv3_input[] = { 290050688,290151039 };
-int layer4_block2_conv3_output[] = { 290151040,290552447 };
-int layer4_block2_bn3_input[] = { 290552448,290953855 };
-int layer4_block2_bn3_output[] = { 290953856,291355263 };
diff --git a/CNNDemo/ResNet50WeightShape.cc b/CNNDemo/ResNet50WeightShape.cc
index c64de37..28a5a5c 100644
--- a/CNNDemo/ResNet50WeightShape.cc
+++ b/CNNDemo/ResNet50WeightShape.cc
@@ -296,211 +296,3 @@ int verify_layer3_shape[] = { 1, 1024, 14, 14,  };
 int verify_layer4_shape[] = { 1, 2048, 7, 7,  };
 int verify_avgpool_shape[] = { 1, 2048, 1, 1,  };
 int verify_fc_shape[] = { 1, 1000,  };
-int layer1_block0_conv1_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block0_conv1_output_shape[] = { 64, 56, 56,  };
-int layer1_block0_bn1_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block0_bn1_output_shape[] = { 64, 56, 56,  };
-int layer1_block0_conv2_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block0_conv2_output_shape[] = { 64, 56, 56,  };
-int layer1_block0_bn2_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block0_bn2_output_shape[] = { 64, 56, 56,  };
-int layer1_block0_conv3_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block0_conv3_output_shape[] = { 256, 56, 56,  };
-int layer1_block0_bn3_input_shape[] = { 1, 256, 56, 56,  };
-int layer1_block0_bn3_output_shape[] = { 256, 56, 56,  };
-int layer1_block0_downsample_conv_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block0_downsample_conv_output_shape[] = { 256, 56, 56,  };
-int layer1_block0_downsample_bn_input_shape[] = { 1, 256, 56, 56,  };
-int layer1_block0_downsample_bn_output_shape[] = { 256, 56, 56,  };
-int layer1_block1_conv1_input_shape[] = { 1, 256, 56, 56,  };
-int layer1_block1_conv1_output_shape[] = { 64, 56, 56,  };
-int layer1_block1_bn1_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block1_bn1_output_shape[] = { 64, 56, 56,  };
-int layer1_block1_conv2_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block1_conv2_output_shape[] = { 64, 56, 56,  };
-int layer1_block1_bn2_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block1_bn2_output_shape[] = { 64, 56, 56,  };
-int layer1_block1_conv3_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block1_conv3_output_shape[] = { 256, 56, 56,  };
-int layer1_block1_bn3_input_shape[] = { 1, 256, 56, 56,  };
-int layer1_block1_bn3_output_shape[] = { 256, 56, 56,  };
-int layer1_block2_conv1_input_shape[] = { 1, 256, 56, 56,  };
-int layer1_block2_conv1_output_shape[] = { 64, 56, 56,  };
-int layer1_block2_bn1_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block2_bn1_output_shape[] = { 64, 56, 56,  };
-int layer1_block2_conv2_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block2_conv2_output_shape[] = { 64, 56, 56,  };
-int layer1_block2_bn2_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block2_bn2_output_shape[] = { 64, 56, 56,  };
-int layer1_block2_conv3_input_shape[] = { 1, 64, 56, 56,  };
-int layer1_block2_conv3_output_shape[] = { 256, 56, 56,  };
-int layer1_block2_bn3_input_shape[] = { 1, 256, 56, 56,  };
-int layer1_block2_bn3_output_shape[] = { 256, 56, 56,  };
-int layer2_block0_conv1_input_shape[] = { 1, 256, 56, 56,  };
-int layer2_block0_conv1_output_shape[] = { 128, 56, 56,  };
-int layer2_block0_bn1_input_shape[] = { 1, 128, 56, 56,  };
-int layer2_block0_bn1_output_shape[] = { 128, 56, 56,  };
-int layer2_block0_conv2_input_shape[] = { 1, 128, 56, 56,  };
-int layer2_block0_conv2_output_shape[] = { 128, 28, 28,  };
-int layer2_block0_bn2_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block0_bn2_output_shape[] = { 128, 28, 28,  };
-int layer2_block0_conv3_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block0_conv3_output_shape[] = { 512, 28, 28,  };
-int layer2_block0_bn3_input_shape[] = { 1, 512, 28, 28,  };
-int layer2_block0_bn3_output_shape[] = { 512, 28, 28,  };
-int layer2_block0_downsample_conv_input_shape[] = { 1, 256, 56, 56,  };
-int layer2_block0_downsample_conv_output_shape[] = { 512, 28, 28,  };
-int layer2_block0_downsample_bn_input_shape[] = { 1, 512, 28, 28,  };
-int layer2_block0_downsample_bn_output_shape[] = { 512, 28, 28,  };
-int layer2_block1_conv1_input_shape[] = { 1, 512, 28, 28,  };
-int layer2_block1_conv1_output_shape[] = { 128, 28, 28,  };
-int layer2_block1_bn1_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block1_bn1_output_shape[] = { 128, 28, 28,  };
-int layer2_block1_conv2_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block1_conv2_output_shape[] = { 128, 28, 28,  };
-int layer2_block1_bn2_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block1_bn2_output_shape[] = { 128, 28, 28,  };
-int layer2_block1_conv3_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block1_conv3_output_shape[] = { 512, 28, 28,  };
-int layer2_block1_bn3_input_shape[] = { 1, 512, 28, 28,  };
-int layer2_block1_bn3_output_shape[] = { 512, 28, 28,  };
-int layer2_block2_conv1_input_shape[] = { 1, 512, 28, 28,  };
-int layer2_block2_conv1_output_shape[] = { 128, 28, 28,  };
-int layer2_block2_bn1_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block2_bn1_output_shape[] = { 128, 28, 28,  };
-int layer2_block2_conv2_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block2_conv2_output_shape[] = { 128, 28, 28,  };
-int layer2_block2_bn2_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block2_bn2_output_shape[] = { 128, 28, 28,  };
-int layer2_block2_conv3_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block2_conv3_output_shape[] = { 512, 28, 28,  };
-int layer2_block2_bn3_input_shape[] = { 1, 512, 28, 28,  };
-int layer2_block2_bn3_output_shape[] = { 512, 28, 28,  };
-int layer2_block3_conv1_input_shape[] = { 1, 512, 28, 28,  };
-int layer2_block3_conv1_output_shape[] = { 128, 28, 28,  };
-int layer2_block3_bn1_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block3_bn1_output_shape[] = { 128, 28, 28,  };
-int layer2_block3_conv2_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block3_conv2_output_shape[] = { 128, 28, 28,  };
-int layer2_block3_bn2_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block3_bn2_output_shape[] = { 128, 28, 28,  };
-int layer2_block3_conv3_input_shape[] = { 1, 128, 28, 28,  };
-int layer2_block3_conv3_output_shape[] = { 512, 28, 28,  };
-int layer2_block3_bn3_input_shape[] = { 1, 512, 28, 28,  };
-int layer2_block3_bn3_output_shape[] = { 512, 28, 28,  };
-int layer3_block0_conv1_input_shape[] = { 1, 512, 28, 28,  };
-int layer3_block0_conv1_output_shape[] = { 256, 28, 28,  };
-int layer3_block0_bn1_input_shape[] = { 1, 256, 28, 28,  };
-int layer3_block0_bn1_output_shape[] = { 256, 28, 28,  };
-int layer3_block0_conv2_input_shape[] = { 1, 256, 28, 28,  };
-int layer3_block0_conv2_output_shape[] = { 256, 14, 14,  };
-int layer3_block0_bn2_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block0_bn2_output_shape[] = { 256, 14, 14,  };
-int layer3_block0_conv3_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block0_conv3_output_shape[] = { 1024, 14, 14,  };
-int layer3_block0_bn3_input_shape[] = { 1, 1024, 14, 14,  };
-int layer3_block0_bn3_output_shape[] = { 1024, 14, 14,  };
-int layer3_block0_downsample_conv_input_shape[] = { 1, 512, 28, 28,  };
-int layer3_block0_downsample_conv_output_shape[] = { 1024, 14, 14,  };
-int layer3_block0_downsample_bn_input_shape[] = { 1, 1024, 14, 14,  };
-int layer3_block0_downsample_bn_output_shape[] = { 1024, 14, 14,  };
-int layer3_block1_conv1_input_shape[] = { 1, 1024, 14, 14,  };
-int layer3_block1_conv1_output_shape[] = { 256, 14, 14,  };
-int layer3_block1_bn1_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block1_bn1_output_shape[] = { 256, 14, 14,  };
-int layer3_block1_conv2_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block1_conv2_output_shape[] = { 256, 14, 14,  };
-int layer3_block1_bn2_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block1_bn2_output_shape[] = { 256, 14, 14,  };
-int layer3_block1_conv3_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block1_conv3_output_shape[] = { 1024, 14, 14,  };
-int layer3_block1_bn3_input_shape[] = { 1, 1024, 14, 14,  };
-int layer3_block1_bn3_output_shape[] = { 1024, 14, 14,  };
-int layer3_block2_conv1_input_shape[] = { 1, 1024, 14, 14,  };
-int layer3_block2_conv1_output_shape[] = { 256, 14, 14,  };
-int layer3_block2_bn1_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block2_bn1_output_shape[] = { 256, 14, 14,  };
-int layer3_block2_conv2_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block2_conv2_output_shape[] = { 256, 14, 14,  };
-int layer3_block2_bn2_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block2_bn2_output_shape[] = { 256, 14, 14,  };
-int layer3_block2_conv3_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block2_conv3_output_shape[] = { 1024, 14, 14,  };
-int layer3_block2_bn3_input_shape[] = { 1, 1024, 14, 14,  };
-int layer3_block2_bn3_output_shape[] = { 1024, 14, 14,  };
-int layer3_block3_conv1_input_shape[] = { 1, 1024, 14, 14,  };
-int layer3_block3_conv1_output_shape[] = { 256, 14, 14,  };
-int layer3_block3_bn1_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block3_bn1_output_shape[] = { 256, 14, 14,  };
-int layer3_block3_conv2_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block3_conv2_output_shape[] = { 256, 14, 14,  };
-int layer3_block3_bn2_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block3_bn2_output_shape[] = { 256, 14, 14,  };
-int layer3_block3_conv3_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block3_conv3_output_shape[] = { 1024, 14, 14,  };
-int layer3_block3_bn3_input_shape[] = { 1, 1024, 14, 14,  };
-int layer3_block3_bn3_output_shape[] = { 1024, 14, 14,  };
-int layer3_block4_conv1_input_shape[] = { 1, 1024, 14, 14,  };
-int layer3_block4_conv1_output_shape[] = { 256, 14, 14,  };
-int layer3_block4_bn1_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block4_bn1_output_shape[] = { 256, 14, 14,  };
-int layer3_block4_conv2_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block4_conv2_output_shape[] = { 256, 14, 14,  };
-int layer3_block4_bn2_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block4_bn2_output_shape[] = { 256, 14, 14,  };
-int layer3_block4_conv3_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block4_conv3_output_shape[] = { 1024, 14, 14,  };
-int layer3_block4_bn3_input_shape[] = { 1, 1024, 14, 14,  };
-int layer3_block4_bn3_output_shape[] = { 1024, 14, 14,  };
-int layer3_block5_conv1_input_shape[] = { 1, 1024, 14, 14,  };
-int layer3_block5_conv1_output_shape[] = { 256, 14, 14,  };
-int layer3_block5_bn1_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block5_bn1_output_shape[] = { 256, 14, 14,  };
-int layer3_block5_conv2_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block5_conv2_output_shape[] = { 256, 14, 14,  };
-int layer3_block5_bn2_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block5_bn2_output_shape[] = { 256, 14, 14,  };
-int layer3_block5_conv3_input_shape[] = { 1, 256, 14, 14,  };
-int layer3_block5_conv3_output_shape[] = { 1024, 14, 14,  };
-int layer3_block5_bn3_input_shape[] = { 1, 1024, 14, 14,  };
-int layer3_block5_bn3_output_shape[] = { 1024, 14, 14,  };
-int layer4_block0_conv1_input_shape[] = { 1, 1024, 14, 14,  };
-int layer4_block0_conv1_output_shape[] = { 512, 14, 14,  };
-int layer4_block0_bn1_input_shape[] = { 1, 512, 14, 14,  };
-int layer4_block0_bn1_output_shape[] = { 512, 14, 14,  };
-int layer4_block0_conv2_input_shape[] = { 1, 512, 14, 14,  };
-int layer4_block0_conv2_output_shape[] = { 512, 7, 7,  };
-int layer4_block0_bn2_input_shape[] = { 1, 512, 7, 7,  };
-int layer4_block0_bn2_output_shape[] = { 512, 7, 7,  };
-int layer4_block0_conv3_input_shape[] = { 1, 512, 7, 7,  };
-int layer4_block0_conv3_output_shape[] = { 2048, 7, 7,  };
-int layer4_block0_bn3_input_shape[] = { 1, 2048, 7, 7,  };
-int layer4_block0_bn3_output_shape[] = { 2048, 7, 7,  };
-int layer4_block0_downsample_conv_input_shape[] = { 1, 1024, 14, 14,  };
-int layer4_block0_downsample_conv_output_shape[] = { 2048, 7, 7,  };
-int layer4_block0_downsample_bn_input_shape[] = { 1, 2048, 7, 7,  };
-int layer4_block0_downsample_bn_output_shape[] = { 2048, 7, 7,  };
-int layer4_block1_conv1_input_shape[] = { 1, 2048, 7, 7,  };
-int layer4_block1_conv1_output_shape[] = { 512, 7, 7,  };
-int layer4_block1_bn1_input_shape[] = { 1, 512, 7, 7,  };
-int layer4_block1_bn1_output_shape[] = { 512, 7, 7,  };
-int layer4_block1_conv2_input_shape[] = { 1, 512, 7, 7,  };
-int layer4_block1_conv2_output_shape[] = { 512, 7, 7,  };
-int layer4_block1_bn2_input_shape[] = { 1, 512, 7, 7,  };
-int layer4_block1_bn2_output_shape[] = { 512, 7, 7,  };
-int layer4_block1_conv3_input_shape[] = { 1, 512, 7, 7,  };
-int layer4_block1_conv3_output_shape[] = { 2048, 7, 7,  };
-int layer4_block1_bn3_input_shape[] = { 1, 2048, 7, 7,  };
-int layer4_block1_bn3_output_shape[] = { 2048, 7, 7,  };
-int layer4_block2_conv1_input_shape[] = { 1, 2048, 7, 7,  };
-int layer4_block2_conv1_output_shape[] = { 512, 7, 7,  };
-int layer4_block2_bn1_input_shape[] = { 1, 512, 7, 7,  };
-int layer4_block2_bn1_output_shape[] = { 512, 7, 7,  };
-int layer4_block2_conv2_input_shape[] = { 1, 512, 7, 7,  };
-int layer4_block2_conv2_output_shape[] = { 512, 7, 7,  };
-int layer4_block2_bn2_input_shape[] = { 1, 512, 7, 7,  };
-int layer4_block2_bn2_output_shape[] = { 512, 7, 7,  };
-int layer4_block2_conv3_input_shape[] = { 1, 512, 7, 7,  };
-int layer4_block2_conv3_output_shape[] = { 2048, 7, 7,  };
-int layer4_block2_bn3_input_shape[] = { 1, 2048, 7, 7,  };
-int layer4_block2_bn3_output_shape[] = { 2048, 7, 7,  };
diff --git a/CNNDemo/Resnet50.py b/CNNDemo/Resnet50.py
index 8c97fbe..226de14 100644
--- a/CNNDemo/Resnet50.py
+++ b/CNNDemo/Resnet50.py
@@ -503,9 +503,9 @@ resnet50.layer4._modules['2'].conv3.register_forward_hook(lambda m, i, o: hook_p
 
 
 
-for batch_idx, (data, target) in enumerate(val_loader):
-    out = resnet50(data)
-    break
+# for batch_idx, (data, target) in enumerate(val_loader):
+#     out = resnet50(data)
+#     break