def METHOD_NAME(self, destination=None, prefix="", keep_vars=False):
    state = ConvTBC.METHOD_NAME(self, destination, prefix, keep_vars=keep_vars)
    # don't store redundant _linearized_weight in checkpoints
    if prefix + "_linearized_weight" in state:
        del state[prefix + "_linearized_weight"]
    return state