(self)
| 127 | self.layer.bn_param.decay = 1 - refine_value(self.op.attributes.get('momentum', 0.9)) |
| 128 | |
| 129 | def parse(self): |
| 130 | super(BatchNormalization, self).parse() |
| 131 | channel = len(self.layer.blobs[3].data) |
| 132 | for i in range(4): |
| 133 | self.layer.blobs[i].shape.ClearField('dim') |
| 134 | self.layer.blobs[i].shape.dim.extend([1, channel, 1, 1]) |
| 135 | return self.layer |
| 136 | |
| 137 | |
| 138 | @register_class |