ptrblck:

Could you remove the direct loading of the obs_model and use the recommended workflow of creating the model instance and loading its state_dict afterwards?


obs_model is actually not a trained model with a state_dict and is a D.normal.Normal distribution object, but I created it anew separately on the CPU anyway and got the same result:

image1288×932 344 KB

Because of a .csv of data the code depends on, I’m unfortunately unable to presently share a minimal independent code snippet to reproduce this issue. Is there a way I could upload a .csv file?
Edit: I’ll put the .csv up on Github and call the link.