Fix missing `cin` attr for fc nodes (#17)
This commit is contained in:
Родитель
a3b72532b7
Коммит
273cd2d4b6
|
@ -62,7 +62,7 @@ class KernelDetector:
|
|||
if "split_dim" in attr:
|
||||
kernel["split_dim"] = attr["split_dim"]
|
||||
|
||||
if len(input_shape) == 1:
|
||||
if len(input_shape) >= 1:
|
||||
if len(input_shape[0]) == 4:
|
||||
kernel["inputh"] = input_shape[0][1]
|
||||
kernel["inputw"] = input_shape[0][2]
|
||||
|
|
Загрузка…
Ссылка в новой задаче