search

keisen / tf-keras-vis

Neural network visualization toolkit for tf.keras
https://keisen.github.io/tf-keras-vis-docs/
MIT License
311 stars 45 forks source link

It does not work with models with custom layers. #90

Closed davidjones1 closed 1 year ago

davidjones1 commented 2 years ago

@keisen, Good library and clean code. Thanks

I have a very simple model for testing a custom convolutional layer. Your code does not work for all the visualizations. Two main errors: 1- it needs to have a model to be cloned. It can be easily fixed by overriding get_config. (Somehow) 2- But the second error is quite a big issue. None of the visualization works for a model with custom layers. image

I could make the conv layers visualization work (and also dense) but the rest: no success. Does you code work only with pre-trained models?

keisen commented 1 year ago

Hi, @davidjones1 .

Does you code work only with pre-trained models?

No, it also works for custom models. Could you submit your code and the full stack of the error message?

davidjones1 commented 1 year ago

IT does not work unfortunately. I would not share the code with you due to intellectual property.

On Monday, November 7, 2022 at 10:53:16 PM EST, Yasuhiro Kubota ***@***.***> wrote:

Hi, @davidjones1 .

Does you code work only with pre-trained models?

No, it also works for custom models. Could you submit your code and the full stack of the error message?

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>

keisen commented 1 year ago

Hi, @davidjones1 .

I'm going to close this issue because there is no cue enough to understand it for now. Please feel free to reopen if you get the code to reproduce the error or face it again on shareable code.

Thanks!