Method set_attr

ppq/parser/caffe/caffe_export_utils.py:123–127

(self)

@register_class
class BatchNormalization(CaffeOpExporter):
    def set_attr(self):
        self.layer.bn_param.moving_average = bool(refine_value(self.op.attributes.get('training_mode', 0)))
        self.layer.bn_param.var_eps = refine_value(self.op.attributes.get('epsilon', 1e-05))
        if self.layer.bn_param.moving_average:
            self.layer.bn_param.decay = 1 - refine_value(self.op.attributes.get('momentum', 0.9))

    def parse(self):
        super(BatchNormalization, self).parse()