Check all model in param groups

[lgvaz]

From icevision:

def check_all_model_params_in_groups2(model: nn.Module, param_groups: List[List[nn.Parameter]]):
    num_params = len([param for group in param_groups for param in group])
    num_params_expected = len(list(model.parameters()))

    if num_params != num_params_expected:
        raise RuntimeError(
            f"{num_params_expected} params in model but only {num_params} "
            "in parameter group"
        )