conv1d_shapes = [
    (1, 192, 56 * 56, 128, 192, 1 * 1, 1, 1, 0, 1, 1), 
    (1, 128, 56 * 56, 256, 128, 3 * 3, 1, 1, 1, 1, 1), 
    (1, 512, 28 * 28, 256, 512, 1 * 1, 1, 1, 0, 1, 1), 
    (1, 256, 28 * 28, 512, 256, 3 * 3, 1, 1, 1, 1, 1),
    (1, 1024, 196, 512, 1024, 1 * 1, 1, 1, 0, 1, 1), 
    (1, 512, 196, 1024, 512, 3 * 3, 1, 1, 1, 1, 1),
    (1, 1024, 49, 1024, 1024, 3 * 3, 1, 1, 1, 1, 1), 
]
conv1d_shapes = [
    (1, 192, 56 * 56, 128, 192, 1 * 1, 1, 1, 0, 1, 1), 
    (1, 128, 56 * 56, 256, 128, 3 * 3, 1, 1, 1, 1, 1), 
    (1, 512, 28 * 28, 256, 512, 1 * 1, 1, 1, 0, 1, 1), 
    (1, 256, 28 * 28, 512, 256, 3 * 3, 1, 1, 1, 1, 1),
    (1, 1024, 196, 512, 1024, 1 * 1, 1, 1, 0, 1, 1), 
    (1, 512, 196, 1024, 512, 3 * 3, 1, 1, 1, 1, 1),
    (1, 1024, 49, 1024, 1024, 3 * 3, 1, 1, 1, 1, 1), 
]