Method reset

diff2flow/lora.py:50–51

(self)

        raise ValueError("Error: no matching batch found")

    def reset(self):
        self.batch = None

class LoraLinear(torch.nn.Module):
    def __init__(