Skip to content
Snippets Groups Projects
Commit b0cd85cb authored by Neta Zmora's avatar Neta Zmora
Browse files

model_summaries.py: add simple Conv layer annotations to PNG export 

1.	Add basic annotations to Conv layers when generating model PNG diagrams
2.	Refactor: replace dataset_dummy_input with the global utility distiller.get_dummy_input
parent bf1e6a0d
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment