search

openclimatefix / satflow

Satellite Optical Flow with machine learning models
https://satflow.readthedocs.io/en/stable/
MIT License
61 stars 10 forks source link

Try out PipeTransformer #80

Open jacobbieker opened 3 years ago

jacobbieker commented 3 years ago

PyTorch recently came out with this https://github.com/Distributed-AI/PipeTransformer that sped up training transformers and potentially using less GPU resources, or at least train faster.