EvilPsyCHo / Deep-Time-Series-Prediction

Seq2Seq, Bert, Transformer, WaveNet for time series prediction.
564 stars 79 forks source link

Running without Cuda #6

Open RandomForestGump opened 3 years ago

RandomForestGump commented 3 years ago

Running your sample code on CPU without cuda arguments in creating train_dl results in too many values to unpack error while training. Any solution?

fitzgerald0 commented 3 years ago

Running your sample code on CPU without cuda arguments in creating train_dl results in too many values to unpack error while training. Any solution?

maybe,you can change like this:

for j, data in enumerate(train_dl): x,y=data[0],data[1]