neuralop.utils.count_model_params

neuralop.utils.count_model_params(model)[source]

Returns the total number of parameters of a PyTorch model

Notes

One complex number is counted as two parameters (we count real and imaginary parts)’