Fix missing `cin` attr for fc nodes (#17)

This commit is contained in:
kalineid 2021-08-12 10:32:17 +08:00 коммит произвёл GitHub
Родитель a3b72532b7
Коммит 273cd2d4b6
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -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]