latest version of pyro;

When I copy the code in GP Regression after using this block of code:

optimizer = torch.optim.Adam(gpr.parameters(), lr=0.005)

loss_fn = pyro.infer.Trace_ELBO().differentiable_loss

losses = []

num_steps = 2500 if not smoke_test else 2

for i in range(num_steps):

optimizer.zero_grad()

loss = loss_fn(gpr.model, gpr.guide)

loss.backward()

optimizer.step()

losses.append(loss.item())

then I try to plot the predictive posterior distribution, I get this:

I continue to run the following tutorial code for sparse GP ,it gets the right figure as in tutorialâ€¦

I just canâ€™t figure out whatâ€™s going wrongâ€¦