I find the cause of the problem, it seems that block hide the sample sites so SVI can’t pass it’s RNGKey to the hided sample sites.
See this post for the solution.
1 Like
I find the cause of the problem, it seems that block hide the sample sites so SVI can’t pass it’s RNGKey to the hided sample sites.
See this post for the solution.