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)
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)