How to run this code with GPU?

shunyaoshih / TPA-LSTM

Temporal Pattern Attention for Multivariate Time Series Forecasting

699 stars 187 forks source link

Open yangye19960922 opened 5 years ago

yangye19960922 commented 5 years ago

Tried many ways, but logger only shows the code was run by CPU

CalibrationMe commented 4 years ago

Does anyone run the code by GPU successfully ?