def test_set_untrainable_on_dense3(): model = create_dense_model((33, ), 27, 41) set_untrainable(model, 1) assert get_trainable_layers(model) == 1148
def test_set_untrainable_on_dense2(): model = create_dense_model((10, ), 10, 10) set_untrainable(model, 0) assert get_trainable_layers(model) == 0
def test_set_untrainable_on_cnn4(): model = create_cnn_model((28, 28, 1), 10, 10) set_untrainable(model, 0) assert get_trainable_layers(model) == 0
def test_set_untrainable_with_negative_argument(): model = create_cnn_model((28, 28, 1), 10, 10) set_untrainable(model, -1) assert get_trainable_layers(model) == 5761
def test_set_untrainable_on_cnn3(): model = create_cnn_model((28, 28, 1), 10, 10) set_untrainable(model, 4) assert get_trainable_layers(model) == 5761 + 910 + 100