segmentation_models
segmentation_models copied to clipboard
Error when trying to remove dropout layers
Hi, because coreml does not support "class 'efficientnet.model.get_dropout.
I've tried
layers = [l for l in model.layers]
x = layers[0](layers[0].input)
for i in range(1, 68):
layers[i].trainable = False
x = layers[i](x)
for i in range(69, len(layers)):
layers[i].trainable = False
x = layers[i](x)
but this results in
ValueError: A merge layer should be called on a list of inputs.
How do I remove intermediate dropout layers?
hey, I know it's an old threat, but I have a similar problem now.. Did you find a solution to that?