rockerBOO 6d42b95e2b Refactor transforms, fix loss calculations ...

- add full conditional_loss functionality to wavelet loss
- Transforms are separate and abstracted
- Loss now doesn't include LL except the lowest level
  - ll_level_threshold allows you to control the level the ll is
    used in the loss
- band weights can now be passed in
- rectified flow calculations can be bypassed for experimentation
- Fixed alpha to 1.0 with new weighted bands producing lower loss

2025-05-04 18:39:32 -04:00

26 KiB

Raw Blame History

View Raw