Pyro.infer.importance.psis_diagnostic gives errors

I found this issue which is exactly the same as mine:

Apologies, should have searched harder!

As a follow up to that discussion:

  • Since importance.psis_diagnostic cannot be used with pytorch modules, how else can we test for the convergence of the SVI?