Счетчик FLOP для модели keras в python - PullRequest
1 голос
/ 06 января 2020

Как получить число операций с плавающей точкой, выполненных моделью Keras, в python?

Model.summary () предоставляет только количество обучаемых и необучаемых параметров. Может кто-нибудь, пожалуйста, скажите мне, как получить количество флопов для модели керас?

Мое определение модели:

model.add (Dense (32, input_dim = train) .shape [1], активация = 'relu'))

model.add (Плотный (len (label_list), активация = 'softmax'))

для меня,

train_shape [1] = 128 label_list = 60

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...