1d regression example

[subhashis]

I was trying to run the 1d regression notebook straight out of the box as provided. Ran into this issue of size mismatch. Any help would be appreciated. Thank you

ValueError Traceback (most recent call last)

in ()
32 optim.zero_grad()
33 y_pred, kl, loss, mse_loss, y_std = model(context_x, context_y, target_x,
---> 34 target_y)
35 loss.backward()
36 optim.step()

ValueError: not enough values to unpack (expected 5, got 3)

[wassname]

What branch/commit are you on? You will need to make sure the repo is on the same commit that you got the notebook from.

If you've checked all that, it might be that I changed the output of the model and haven't updated the script. It should be fairly easy for you to update that line.. It's been a while since I've worked on this, I would need more info to be able to help.