Getting the parameters and their names of scVI

Hi,
I used to use model.named_parameters in pytorch to get the model’s parameters and their names. I found the analogous in scvi

scvi.module.SCANVAE.named_parameters

but I could not understand the provided example

for name, param in self.named_parameters():
if name in [‘bias’]:
print(param.size())

I am not sure how to use it to get the parameters of the scVI model I am using. For example, I am calling the scVI model ‘MODEL’. How can I get the ‘MODEL’ parameters and their names?

Thanks in advance :slight_smile:

You want to do something like

model = scvi.model.SCVI(adata)
model.module # this is a pytorch module
model.module.named_parameters()