import torch
from typing import Tuple


def get_model_size(model: torch.nn.Module) -> Tuple[int, int]:
    """Return number of model parameters."""

    num_total_params = sum([p.numel() for p in model.parameters()])
    num_train_params = sum([p.numel() for p in model.parameters() if p.requires_grad])
    return num_total_params, num_train_params