search

openclimatefix / graph_weather

PyTorch implementation of Ryan Keisler's 2022 "Forecasting Global Weather with Graph Neural Networks" paper (https://arxiv.org/abs/2202.07575)
MIT License
188 stars 47 forks source link

Update to take advantage of PyG 2.2.0 #53

Closed jacobbieker closed 10 months ago

jacobbieker commented 1 year ago

Detailed Description

https://github.com/pyg-team/pytorch_geometric/releases/tag/2.2.0

Context

New version has improvements for working on larger-than-memory graphs, etc. which might be helpful here, and also the speed ups that should be included.

Possible Implementation