Function count_parameters

utils/interp.py:32–33

(model)

def param_count(module):
    def count_parameters(model):
        return sum(p.numel() for p in model.parameters() if p.requires_grad)
    
    total_params = count_parameters(module)
    output = [f'Total model parameters: {total_params:,}', '---------------------------']