예제 #1
0
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
예제 #2
0
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