Source code for src.utils.count_model_params

import numpy as np

[docs]def count_parameters(model): return np.sum(np.prod(v.size()) for name, v in model.named_parameters() if "auxiliary" not in name)