scVI access the model

Hi I was wondering how I would access the different model layers within the scVI class structure.
In particular I want to extract the weights from the input layer to the first hidden layer.
Has someone an idea how to approach this given a vae object of type scvi.model._scvi.SCVI?

vae = scvi.model.SCVI(...)
# pytorch module

should work!

1 Like

Thank you for your response! I attempted to follow the steps, but I found the documentation to be a bit confusing. Can I rely solely on the PyTorch documentation for further guidance?

So vae.module is a torch nn.Module. Therefore yes I would follow the PyTorch documentation for how to use this

1 Like

Thank you so much for the clarification!