Ejemplos de change_trainable en Python

Ejemplo n.º 1

Archivo: test_finetuning.py Proyecto: Dobatymo/DeepMoji

def test_change_trainable():
    """ change_trainable() changes trainability of layers.
    """
    model = deepmoji_transfer(5, 30)
    change_trainable(model.layers[0], False)
    assert not model.layers[0].trainable
    change_trainable(model.layers[0], True)
    assert model.layers[0].trainable

Ejemplo n.º 2

def test_change_trainable():
    """ change_trainable() changes trainability of layers.
    """
    model = deepmoji_transfer(5, 30)
    change_trainable(model.layers[0], False)
    assert not model.layers[0].trainable
    change_trainable(model.layers[0], True)
    assert model.layers[0].trainable