Function _get_params

TinyCLIP/src/training/main.py:57–62

(model)

def compute_params(model):
    def _get_params(model):
        if model is None:
            return 0
        n_parameters = sum(p.numel()
                           for p in model.parameters() if p.requires_grad)
        return n_parameters

    def _get_buffers(model):
        if model is None: