pytorch_get_n_params_1.py

def get_num_params(model):
  param_count = 0
  for param in model.parameters():
      param_count += np.product(param.data.shape)
  return param_count

pytorch_get_n_params_2.py

def get_num_params(model):
  return reduce(operator.add,
                map(lambda param: reduce(operator.mul, [s for s in param.size()]),
                    model.parameters())))