 
   Conv2D<float> conv5(3,3,384,256,1, new GaussianKernelInitializer<float>(0.1), SAME);
   size_t conv5_top_dims[4];
-  conv4.GetTopsDims({relu4_top_dims}, {conv5_top_dims});
+  conv5.GetTopsDims({relu4_top_dims}, {conv5_top_dims});
 
   Pooling<float> pool5(2, MAX, 2);
   size_t pool5_top_dims[4];



int main() {
  test_alexnet_cpu();
  //test_alexnet_gpu();
}


  Conv2D<float> conv5(3,3,384,256,1, new GaussianKernelInitializer<float>(0.1), SAME);
  size_t conv5_top_dims[4];
  conv5.GetTopsDims({relu4_top_dims}, {conv5_top_dims});

  Pooling<float> pool5(2, MAX, 2);
  size_t pool5_top_dims[4];
