Function _count_params

model/splat_model.py:22–23

(module)

def _count_params(module):
    return sum(p.numel() for p in module.parameters() if p.requires_grad)

class SplatModel(StaticEncoder):
    def __init__(self, opt: Options, **model_kwargs):