def generate_units_size(network_size, layer_size):
    assert (network_size > 0), 'Network size invalid'
    assert (layer_size > 0), 'Layer size invalid'
    num_units = []
    for i in range(network_size):
        num_units.append(int(layer_size))
        if (layer_size != 2):
            layer_size /= 2
    return num_units
def generate_units_size(network_size, layer_size):
    assert (network_size > 0), 'Network size invalid'
    assert (layer_size > 0), 'Layer size invalid'
    num_units = []
    for i in range(network_size):
        num_units.append(int(layer_size))
        if (layer_size != 2):
            layer_size /= 2
    return num_units