Skip to content
Snippets Groups Projects
Commit 093a2e17 authored by Lev Zlotnik's avatar Lev Zlotnik Committed by Neta Zmora
Browse files

Model drawing: allow rendering models w/o specifying the dataset (#294)

This commit allows us to draw diagrams of models, even if we don't support the specific dataset used by the model.  All you need is to specify the dimensions of the model inputs.
* Added `input_shape` argument to `draw_img_classifier_to_file`
* updated docstring
parent 2cab7741
No related branches found
No related tags found
No related merge requests found
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment