Method __init__

k_diffusion/models/flops.py:21–22

(self)

class FlopCounter:
    def __init__(self):
        self.ops = []

    def op(self, op, *args, **kwargs):
        self.ops.append((op, args, kwargs))