openclimatefix / metnet

PyTorch Implementation of Google Research's MetNet and MetNet-2

MIT License

237 stars 49 forks source link

Add FiLM layers to ConvLSTM #32

Open jacobbieker opened 2 years ago

jacobbieker commented 2 years ago

Pull Request

Description

Adds the FiLM layers to the ConvLSTM layers for MetNet-2, to better match the paper diagram.

Fixes #31

How Has This Been Tested?

Unit test

[ ] Yes

Checklist:

[ ] My code follows OCF's coding style guidelines
[ ] I have performed a self-review of my own code
[ ] I have made corresponding changes to the documentation
[ ] I have added tests that prove my fix is effective or that my feature works
[ ] I have checked my code and corrected any misspellings