def METHOD_NAME(self):
    # Final layer initializations must be 0
    if self.norm_position == 'pre':
        self.module[-1].weight.data *= 0
    elif self.norm_position == 'post':
        self.module[-1].weight.data *= 0
        self.module[-1].bias.data *= 0