AttributeError: 'SCVI' object has no attribute 'named_parameters'

Hello,
I am trying to get the parameters of the scVI model (using the implementation here Unsupervised surgery pipeline with SCVI — scArches documentation)
by named_parameters as I used to do with the standard neural networks implemented using pytorch but I get this error

AttributeError: ‘SCVI’ object has no attribute ‘named_parameters’

Any Suggestions?

Perhaps you can try

vae = scvi.model.SCVI(adata)
# This is a PyTorch nn.Module
vae.module