Does a scvi model has something like pytorchs state_dict?

Help scvi-tools
1

Hey :slight_smile:
I am using an external package with my scvi model and this package (originally written for pytorch models) tries to access the model.state_dict. i.e this dictionary where all the weights and so on are saved. With my scvi model I get the Error :

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

Is there something similar to pytorchs state_dict for scvi models ?
Thanks a lot for your help!

2

Hi, the module attribute of SCVI is a regular PyTorch module, so you can access it through model.module.state_dict.

3

Ahhh, awesome! Thank you :slight_smile: