openclimatefix / metnet

PyTorch Implementation of Google Research's MetNet and MetNet-2
MIT License
231 stars 47 forks source link

Add UK National Training Script #42

Closed jacobbieker closed 1 year ago

jacobbieker commented 1 year ago

Pull Request

Description

Adds training script for the UK National forecast

Fixes #30

How Has This Been Tested?

Unit tests/Training

Checklist:

codecov-commenter commented 1 year ago

Codecov Report

Merging #42 (1ee70a7) into main (f4ab8ea) will decrease coverage by 0.10%. The diff coverage is 94.59%.

@@            Coverage Diff             @@
##             main      #42      +/-   ##
==========================================
- Coverage   84.85%   84.74%   -0.11%     
==========================================
  Files          15       16       +1     
  Lines         581      590       +9     
==========================================
+ Hits          493      500       +7     
- Misses         88       90       +2     
Impacted Files Coverage Δ
metnet/models/metnet.py 93.75% <85.71%> (-1.34%) :arrow_down:
metnet/models/metnet2.py 91.08% <87.50%> (-1.77%) :arrow_down:
metnet/layers/ConditionTime.py 86.95% <100.00%> (ø)
metnet/layers/ConditionWithTimeMetNet2.py 100.00% <100.00%> (ø)
metnet/layers/ConvGRU.py 93.22% <100.00%> (ø)
metnet/layers/ConvLSTM.py 96.62% <100.00%> (ø)

:mega: We’re building smart automated test selection to slash your CI/CD build times. Learn more