conv3d_shapes = [
    # yolo
    # (1, 3, 16, 112, 112, 64, 3, 7, 1, 2, 3, 1, 1), 
    # (1, 64, 16, 28, 28, 192, 64, 3, 1, 1, 1, 1, 1),  
    # (1, 192, 4, 28, 28, 128, 192, 1, 1, 1, 0, 1, 1), 
    # (1, 128, 4, 28, 28, 256, 128, 3, 1, 1, 1, 1, 1),  
    # (1, 512, 4, 14, 14, 256, 512, 1, 1, 1, 0, 1, 1),  
    # (1, 256, 4, 14, 14, 512, 256, 3, 1, 1, 1, 1, 1), 
    # (1, 1024, 4, 7, 7, 512, 1024, 1, 1, 1, 0, 1, 1),  
    # (1, 512, 4, 7, 7, 1024, 512, 3, 1, 1, 1, 1, 1),  

    # resnet-18
    # N, C,   D,  H,   W,   _, K, _,   kernel_size, stride=1,  padding=0, dilation=1, groups=1
    # skip the first layer
    # ( 1, 3,   16, 112, 112, 64,  7,           (1, 2, 2), 3,         1,          1),
    # conv2_x
    ( 1, 64,  8,  28,  28,  64,  64, 3, 1,          1,         1,         1,          1),
    # ( 1, 64,  8,  28,  28,  64,  3,           1,         1,         1,          1),
    # conv3_x
    ( 1, 64,  8,  28,  28,  128, 64, 3, 1,          2,         1,         1,          1),
    ( 1, 128,  4,  14,  14,  128, 128, 3, 1,          1,         1,         1,          1),
    ( 1, 64,  8,  28,  28,  128, 64, 1, 1,          2,         1,         1,          1),

    # ( 1, 128,  4,  14,  14,  128, 3,           1,         1,         1,          1),
    # ( 1, 128,  4,  14,  14,  128, 3,           1,         1,         1,          1),
    # conv4_x
    ( 1, 128, 4,  14,  14,  256, 128, 3, 1,          2,         1,         1,          1),
    ( 1, 256, 2,  7,  7,  256, 256, 3, 1,          1,         1,         1,          1),
    ( 1, 128, 4,  14,  14,  256, 128, 1, 1,          2,         1,         1,          1),

    # ( 1, 256, 2,  7,  7,  256, 3,           1,         1,         1,          1),
    # ( 1, 256, 2,  7,  7,  256, 3,           1,         1,         1,          1),
    # conv5_x
    ( 1, 256, 2,  7,   7,   512, 256, 3, 1,          2,         1,         1,          1),
    ( 1, 512, 1,  3,   3,   512, 512, 3, 1,          1,         1,         1,          1),
    ( 1, 256, 2,  7,   7,   512, 256, 1, 1,          2,         1,         1,          1),

    # ( 1, 512, 1,  3,   3,   512, 3,           1,         1,         1,          1),
    # ( 1, 512, 1,  3,   3,   512, 3,           1,         1,         1,          1),
]