My loss doens’t go to inf. The loss seems similar to that of regular ADVI (which I think is expected, no?
To your first statement about the data_shape thing, let me make sure I understand you.
it sounds like you’re saying I’m using the correct syntax here, and I don’t need to make a change, right?