def count_params(self):
'''Returns the total number of floats (or ints)
composing the weights of the layer.
'''
if not self.built:
if self.__class__.__name__ == 'Sequential':
self.build()
else:
raise Exception('You tried to call `count_params` on ' +
self.name + ', but the layer isn\'t built. '
'You can build it manually via: `' +
self.name + '.build(batch_input_shape)`.')
return sum([K.count_params(p) for p in self.weights])
评论列表
文章目录