wrong variable

[ben10ben]

Hi, there is an issue with iTransformer.py in line 210 and the other iTransformer implementations as well.

assert targets.shape == pred_list.shape

should be changed to

assert target.shape == pred.shape

[lucidrains]

@ben10ben hey! thanks for catching this, should be fixed

have you tried any of the variants? for your work or otherwise?

[ben10ben]

Tried the iTransformer1d with RevIn for the electricity dataset. My implementation is similar to the initial paper but uses 10% more time series.

lookback: 96
pred_length: 96

iTransformer paper MSE: 0.148

train MSE: 0.116, Epoch=10
test MSE: 0.169

Did not get around to doing hyperparameter tuning, but I will hopefully be able to publish a paper beginning of next year.

[lucidrains]

nice! looks great! thank you for relaying 💯