I am using pyro 0.21 . Some examples use max_plate_nesting as an argument to SVI to to Elbo. However, I get an error when executing:
from pyro.infer import SVI, JitTraceEnum_ELBO, TraceEnum_ELBO
Elbo = JitTraceEnum_ELBO if args.jit else TraceEnum_ELBO
elbo = Elbo(max_plate_nesting=2) # orig
This code comes from the LDA example included in lda.py, found in the examples/ folder of the pyro distribution. If it is deprecated, what does one use instead? Thanks.