ai2cm / fv3net

explore the FV3 data for parameterization
MIT License
16 stars 3 forks source link

fix reset state bug occuring every batch #2331

Closed AnnaKwa closed 1 year ago

AnnaKwa commented 1 year ago

The previous PR to enable training over multiple time series had a bug where the state was reset at the start of each batch. This fixes it so it only resets before the first batch of data in each time series is loaded.