scVI access the model

Help scvi-tools
, ,
1

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?

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

should work!

1 Like
3

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?

4

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

1 Like
5

Thank you so much for the clarification!