Function params_count

graphgym/utils/comp_budget.py:7–15

Computes the number of parameters. Args: model (nn.Module): PyTorch model

(model)

def params_count(model):
    '''
    Computes the number of parameters.

    Args:
        model (nn.Module): PyTorch model

    '''
    return sum([p.numel() for p in model.parameters()])


def get_stats():