search

wassname / attentive-neural-processes

implementing "recurrent attentive neural processes" to forecast power usage (w. LSTM baseline, MCDropout)
Apache License 2.0
90 stars 23 forks source link

1d regression example #4

Open subhashis opened 2 years ago

subhashis commented 2 years ago

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 commented 2 years ago

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.