def set_bn_weight_decay(self, bn_weight_decay: bool):
    assert (type(bn_weight_decay) == bool)
    self.bn_weight_decay = bn_weight_decay
    return self
def set_bn_weight_decay(self, bn_weight_decay: bool):
    assert (type(bn_weight_decay) == bool)
    self.bn_weight_decay = bn_weight_decay
    return self