def getdims(x=(10000, 200)):
    """
    return the dims for network
    """
    assert len(x) == 2
    n_sample = x[0]
    if n_sample > 20000:
        dims = [x[-1], 128, 32]
    elif n_sample > 10000:
        dims = [x[-1], 64, 32]
    elif n_sample > 5000:
        dims = [x[-1], 32, 16]
    elif n_sample > 2000:
        dims = [x[-1], 128]
    elif n_sample > 500:
        dims = [x[-1], 64]
    else:
        dims = [x[-1], 16]
    return dims
