search

luke14free / pm-prophet

GAM timeseries modeling with auto-changepoint detection. Inspired by Facebook Prophet and implemented in PyMC3
325 stars 42 forks source link

Is GPU included? #20

Closed JafferWilson closed 4 years ago

JafferWilson commented 4 years ago

Hello Developers of this wonderful repository, I have tried samples. It looks good. But I haven't got any settings for the usage of GPU. It works with the CPU only. What if I am willing to go with the GPUs in my system for the training? Can you help and guide me?

luke14free commented 4 years ago

Hi there, this strongly depends on theano and pymc3 under the hood, which currently relies on Theano (which is discontinued). In theory I think you can use AVDI with GPU, but refer to pymc3 and theano documentations for the setup.