openclimatefix / metnet

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

Change to one lead time per forward pass #20

Closed jacobbieker closed 2 years ago

jacobbieker commented 2 years ago

Pull Request

Description

Changes MetNet and MetNet2 to only predict a single lead time at a time, to predict on all, then need to do multiple forward passes.

How Has This Been Tested?

Unit Tests

Checklist:

codecov-commenter commented 2 years ago

Codecov Report

Merging #20 (f50ad78) into main (f76911f) will increase coverage by 0.13%. The diff coverage is 96.29%.

Impacted file tree graph

@@            Coverage Diff             @@
##             main      #20      +/-   ##
==========================================
+ Coverage   84.89%   85.03%   +0.13%     
==========================================
  Files          15       15              
  Lines         576      568       -8     
==========================================
- Hits          489      483       -6     
+ Misses         87       85       -2     
Impacted Files Coverage Δ
metnet/models/metnet2.py 94.23% <95.83%> (-0.22%) :arrow_down:
metnet/models/metnet.py 94.91% <100.00%> (-0.33%) :arrow_down:
metnet/layers/ConvGRU.py 93.22% <0.00%> (+1.69%) :arrow_up:

Continue to review full report at Codecov.

Legend - Click here to learn more Δ = absolute <relative> (impact), ø = not affected, ? = missing data Powered by Codecov. Last update 7cde48a...f50ad78. Read the comment docs.