spasity function only work on keras sequential model/layer?

[Charlie4zc]

Since i'd like to use sparsity.keras.prune_low_magnitude on my model and failed due to unknown reason, i was wondering whether it only work on keras sequential models?
BTW my keras model has a lot skip connections , so i can only apply on pretrained models（which it works）or sequential 'block's of my model？

[alanchiao]

It currently works on sequential and functional models, though it'll throw an error for subclassed models. It'd be good to update the docs to clarify this also.

Assigning to self since I'm working on initial subclassed model support.

[alanchiao]

Closing now.

If you're interested in subclassed model support, see this issue.