issue 197

This commit is contained in:
Sefik Serengil 2021-03-09 22:16:34 +03:00
parent 0158ae2d2f
commit fee02411b3

View File

@ -486,6 +486,9 @@ def find_input_shape(model):
input_shape = input_shape[0][1:3]
else:
input_shape = input_shape[1:3]
if type(input_shape) == list: #issue 197: some people got array here instead of tuple
input_shape = tuple(input_shape)
return input_shape