search

Arturus / kaggle-web-traffic

1st place solution
MIT License
1.82k stars 667 forks source link

Is there a version which can train the model by CPU? #16

Closed waldstein1983 closed 5 years ago

waldstein1983 commented 6 years ago

Unfortunately, I currently do not have GPU support

Toniiiio commented 6 years ago

If there is no version (which i would expect), maybe an alternative could be of interest for you: https://www.r-bloggers.com/time-series-deep-learning-forecasting-sunspots-with-keras-stateful-lstm-in-r/

Arturus commented 5 years ago

Unfortunately no, because this code depends on CUDA RNN libraries for training. But you can replace tensorflow.contrib.cudnn_rnn.CudnnGRU in a model.py to CPU-friendly variant, such as GRUCell + tf.nn.dynamic_rnn. It should be easy enough.