PyroModule for LSTM

I try to make a SVI of LSTM using PyroModule but it seems not work.
The errors:
File “C:\Users…\Anaconda3\lib\site-packages\torch\nn\modules\”, line 559, in forward self.dropout,, self.bidirectional, self.batch_first)
TypeError: expected Tensor as element 0 in argument 2, but got PyroSample

yes, Ive got this after the newest pytorch upgrade as well. havent figured out why yet.

You mean using old version can solve it?

it worked before 1.4 yes. Otherwise, I can contribute to investigate here, but havent had time yet.

Did you figure this out? Now I need to fix it as well.

It haven’t worked and I am trying to make it using MCMC.