experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix plot_model for PDF

Browse Source
This commit is contained in:
“jaketae” 2020-03-02 06:46:35 +09:00
parent 37637ae605
commit 82a6ba4c50
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/python/keras/utils/vis_utils.py
View File

@ -293,8 +293,9 @@ def plot_model(model,
# Return the image as a Jupyter Image object, to be displayed in-line. # Return the image as a Jupyter Image object, to be displayed in-line.
# Note that we cannot easily detect whether the code is running in a # Note that we cannot easily detect whether the code is running in a
# notebook, and thus we always return the Image if Jupyter is available. # notebook, and thus we always return the Image if Jupyter is available.
try: if extension != 'pdf':
from IPython import display try:
return display.Image(filename=to_file) from IPython import display
except ImportError: return display.Image(filename=to_file)
pass except ImportError:
pass