Bbonev/updated model#35

Draft

bonevbs wants to merge 61 commits intomainfrom

bbonev/updated-model

Commits on Feb 19, 2026

adding index exclusion list support in dali loaders, making index handling more failsafe

azrael417
authored and
bonevbs
committed
some small fixes for non combined dali loader and unannotated files

azrael417
authored and
bonevbs
committed
distringuishing between read_shape and return_shape. read_shape will only be used by the dataloader internally, return_shape can be queried by other modules to determine the size of the returned sa…

azrael417
authored and
bonevbs
committed
fixing a bug where samples per year was not computed correctly

azrael417
authored and
bonevbs
committed
making nicer prints

azrael417
authored and
bonevbs
committed
fixed dataloader indexing

azrael417
authored and
bonevbs
committed
fixed history with noise

azrael417
authored and
bonevbs
committed
setting explicit noise seeds for ensemble serial settings. without this, noise between ensemble members is likely correlated

azrael417
authored and
bonevbs
committed
adding accessor functions for some RNG features when noise is enabled

azrael417
authored and
bonevbs
committed
removing unneccessary imports

azrael417
authored and
bonevbs
committed
removing more imports

azrael417
authored and
bonevbs
committed
removing more imports

azrael417
authored and
bonevbs
committed
resolving rebase conflict
bonevbs
committed
some changes to the model and losses
bonevbs
committed
restoring logic from main regarding noise states
bonevbs
committed
fixing model package
bonevbs
committed
some changes to deterministic trainer
bonevbs
committed
removing the residual training optiona and instead extending loss handler to be able to handle tendencies
bonevbs
committed
updating other configs
bonevbs
committed
added routine to compute spherical bandlimit
bonevbs
committed
resolving rebase conflict
bonevbs
committed
adding energy score loss
bonevbs
committed
removing the random CRPS
bonevbs
committed
resolving rebase conflict
bonevbs
committed
more fixes for energy score loss
bonevbs
committed
resolving rebase conflict
bonevbs
committed
fixing another bug
bonevbs
committed
fixing energy score
bonevbs
committed
cleaning up some losses
bonevbs
committed
implemented improved Sobolev energy score
bonevbs
committed
resolving rebase conflict
bonevbs
committed
fixing bugs in implementations of the energy score
bonevbs
committed
adding stochasticity to vit
bonevbs
committed
adding spectral coherence loss
bonevbs
committed
incorporaitng updates to FCN3 and energy score
bonevbs
committed
small refactor

azrael417
authored and
bonevbs
committed
fixing Pangu and adding tf32 disabling
bonevbs
committed
adding default indexing mode to pangu meshgrid
bonevbs
committed
maknig stochastic FCN3 an option
bonevbs
committed
some bug fixes to spectral CRPS and associated loss tests
bonevbs
committed
updating lm calc

azrael417
authored and
bonevbs
committed
making observations in CRPS test non-constant to mitigate SHT precision issues

azrael417
authored and
bonevbs
committed
streamlining tests

azrael417
authored and
bonevbs
committed
adding energy score

azrael417
authored and
bonevbs
committed
adding disable_tf32
bonevbs
committed
getting rid of debug print
bonevbs
committed
test cleanup

azrael417
authored and
bonevbs
committed
working losses

azrael417
authored and
bonevbs
committed
moving lm weights computation to base loss
bonevbs
committed
working sobolev test

azrael417
authored and
bonevbs
committed
fixing sobolev loss
bonevbs
committed
fixing sobolev loss
bonevbs
committed
adding sobolev loss nan test and removing redundant code

azrael417
authored and
bonevbs
committed
testing only even ensemble sizes

azrael417
authored and
bonevbs
committed
adding relative weight to sobolev energy score loss term and adding loss and grad norm tracking during training

azrael417
authored and
bonevbs
committed
full 0 masking in energy score losses

azrael417
authored and
bonevbs
committed
working energy scores

azrael417
authored and
bonevbs
committed
various cleanups. Factoring out SST imputation
bonevbs
committed
various fixes
bonevbs
committed
fixing imputation and masking in losses
bonevbs
committed
compatibility with newest torch-harmonics
bonevbs
committed