search

hydrogo / rainymotion

Python library for radar-based precipitation nowcasting based on optical flow techniques
MIT License
182 stars 55 forks source link

How I can use GPU with rainymotion? #8

Closed martinscabin closed 2 years ago

martinscabin commented 5 years ago

Hi,

I have a large amount of data right now, and I wonder if there is a way that I can train the rainymotion model with GPU...

hydrogo commented 5 years ago

Hi there, Our rainymotion models work out of the box without any training -- there are only some parameters of optical flow calculation and advection scheme you can adjust by hand or using more complex optimization routines (i.e., differential evolution). At the moment, rainymotion models run pretty fast on CPU, but we do not consider developing the support for computation on GPU.

Our group is also currently working on rainnet -- the deep neural network for precipitation nowcasting that can be trained from scratch on a new data. Unfortunately, we can not ready to share rainnet code and training routine until the submission of the corresponding paper. Thank you for your understanding.

Cheers, Georgy