graphorge.gnn_base_model.model.model_summary.get_n_model_parameters¶

get_n_model_parameters(model)[source]¶

Get number of parameters of PyTorch model.

Parameters:

model (torch.nn.Module) – PyTorch model.

Returns:

n_total_params (int) – Total number of model parameters.
n_train_params (int) – Number of trainable model parameters.
n_nontrain_params (int) – Number of non-trainable model parameters.