| 713 | """ |
| 714 | |
| 715 | def __init__(self, groups=32, epsilon=1e-06, act=None, data_format='channels_last', name=None): #'groupnorm'): |
| 716 | # super(GroupNorm, self).__init__(prev_layer=prev_layer, act=act, name=name) |
| 717 | super().__init__(name, act=act) |
| 718 | self.groups = groups |
| 719 | self.epsilon = epsilon |
| 720 | self.data_format = data_format |
| 721 | |
| 722 | logging.info( |
| 723 | "GroupNorm %s: act: %s" % (self.name, self.act.__name__ if self.act is not None else 'No Activation') |
| 724 | ) |
| 725 | |
| 726 | def build(self, inputs_shape): |
| 727 | # shape = inputs.get_shape().as_list() |