LSTM could not broadcast input array from shape (2302,) into shape (1440,)
ValueError Traceback (most recent call last)
Cell In[59], line 114
111 print(len(test_predict))
113 # Assign predictions to the respective plot segments
114 train_predict_plot[look_back:len(train_predict) + look_back, 0] = train_predict[:, 0]
115 test_predict_plot[len(train_predict) + (look_back * 2) + 1:len(scaled_data) – 1, 0] = test_predict[:, 0]
117 plt.figure(figsize=(10, 6))
My code can’t use gpu to do some training
I’ve built a TCN-LSTM model to do some prediction task. I have already used Cuda in my code. But when I ran them, I found that my cpu was 100% and my gpu didn’t work. I want to use GPU to improve my training speed.