Commit 6f2ef102 authored by rocking's avatar rocking
Browse files

Use proper size

parent e74b1dc8
...@@ -35,12 +35,12 @@ int main() ...@@ -35,12 +35,12 @@ int main()
// Pool shape // Pool shape
ck::index_t N = 2; ck::index_t N = 2;
ck::index_t C = 32; ck::index_t C = 32;
ck::index_t Z = 3; ck::index_t Z = 2;
ck::index_t Y = 3; ck::index_t Y = 2;
ck::index_t X = 3; ck::index_t X = 2;
ck::index_t Di = 31; ck::index_t Di = 30;
ck::index_t Hi = 31; ck::index_t Hi = 30;
ck::index_t Wi = 31; ck::index_t Wi = 30;
ck::index_t window_stride_d = 2; ck::index_t window_stride_d = 2;
ck::index_t window_stride_h = 2; ck::index_t window_stride_h = 2;
ck::index_t window_stride_w = 2; ck::index_t window_stride_w = 2;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment