Commit Graph

9 Commits

Author SHA1 Message Date
rockerBOO
9629853d15 Fix wavelet loss not separating levels. Refactor loss to be spatial 2025-06-05 22:03:52 -04:00
rockerBOO
0af0302c38 Metrics, energy, loss 2025-05-19 19:15:23 -04:00
rockerBOO
984472ca09 Fix metrics 2025-05-04 18:58:54 -04:00
rockerBOO
7be3c5dce1 Refactor SWT to work properly and faster. Add SWT tests 2025-05-04 18:57:53 -04:00
rockerBOO
964bfcb576 Fix padding for small latents. Add DWT tests 2025-05-04 18:57:42 -04:00
rockerBOO
89f0d27a59 Set sigmoid_scale to default 1.0 2025-03-20 15:10:33 -04:00
rockerBOO
d40f5b1e4e Revert "Scale sigmoid to default 1.0"
This reverts commit 8aa126582e.
2025-03-20 15:09:50 -04:00
rockerBOO
8aa126582e Scale sigmoid to default 1.0 2025-03-20 15:09:11 -04:00
rockerBOO
e8b3254858 Add flux_train_utils tests for get get_noisy_model_input_and_timesteps 2025-03-20 15:01:15 -04:00