def count_params(model):
    """returns (total n° of parameters, n° of trainable parameters)"""
    total_params = sum(p.numel() for p in model.parameters())
    trainable_params = sum(p.numel() for p in model.parameters() if p.requires_grad)
    return total_params, trainable_params