Function count_parameters

experiments/exp_era5.py:122–123

(model)

    return hparams

def count_parameters(model):
    return sum(p.numel() for p in model.parameters() if p.requires_grad)

def fit(args):
    positional_encoding_name = args.pe