Replaced the batch sizes with 'None' instead '?'
I have replaced the default text to be shown if batch sizes are not present as 'None' instead of '?' because it is more readable and understandable during the plotting of the model. Furthermore, it avoids the confusion of any error in the model and signifies that the fact that the programmer voluntarily haven't given any batch sizes
This commit is contained in:
parent
3a3d359192
commit
3f4e4cb19d
@ -206,7 +206,7 @@ def model_to_dot(model,
|
||||
if show_shapes:
|
||||
|
||||
def format_shape(shape):
|
||||
return str(shape).replace(str(None), '?')
|
||||
return str(shape).replace(str(None), 'None')
|
||||
|
||||
try:
|
||||
outputlabels = format_shape(layer.output_shape)
|
||||
|
Loading…
x
Reference in New Issue
Block a user