Normalization fixes - #7

Merged

nkern merged 38 commits into

mainfrom

normalization_fixes

May 11, 2026

Commits on May 5, 2026

update to normalization
nkern
committed
2066969
updated data normalization
nkern
committed
30ba844
modified: cosmodiff/utils.py
nkern
committed
8b112c4
modified: cosmodiff/tests/test_utils.py
nkern
committed
93f0c95
modified: cosmodiff/data/config.yaml
nkern
committed
642632b

Commits on May 6, 2026

modified: cosmodiff/data/config.yaml
nkern
committed
ea59d75
fixed training loss for v_prediction
nkern
committed
3db54c8
modified: pyproject.toml
nkern
committed
811014c
added CFG to training; added files in data
nkern
committed
96e2a51
test for continuous conditioning PixArt DIT model
nkern
committed
0afeebb
added EMA weighting to train / checkpoint
nkern
committed
fb0f5dd
modified: cosmodiff/data/config.yaml
nkern
committed
053343d
modified: cosmodiff/data/config.yaml
nkern
committed
c9ae990
modified: scripts/cosmodiff_train.py
nkern
committed
9c33308
modified: cosmodiff/tests/test_optim.py
nkern
committed
12c9af8
modified: cosmodiff/utils.py
nkern
committed
fad1d24

Commits on May 7, 2026

fixed ema update
nkern
committed
e92e5fc
modified: cosmodiff/optim.py
nkern
committed
1925d9b
default ema_update=1, load_ema_snapshot func
nkern
committed
883de50
added min-snr to training loss
nkern
committed
991cbd2
added log-uniform time sampling
nkern
committed
dc78fc3
updated inference script with sampling additions
nkern
committed
f15c730
added ema burn in
nkern
committed
2a97dd7

Commits on May 8, 2026

added transform.py, moved Normalization too
nkern
committed
8560159
updated load_data to handle multiple paths
nkern
committed
70a2151

Commits on May 9, 2026

added support for multiple data files, multiple normalizations
nkern
committed
55b75c7
modified: cosmodiff/utils.py
nkern
committed
ae03be4

Commits on May 10, 2026

modified: cosmodiff/utils.py
nkern
committed
3ac6e8b
modified: cosmodiff/utils.py
nkern
committed
3abf182
pyproject fixes
nkern
committed
8109dfd
updated versioning system
nkern
committed
9e24865

Commits on May 11, 2026

changed rfft2 to fft2
nkern
committed
8508e3f
added scheduler to generate, rm ddim_thin
nkern
committed
87f0abc
added flowmatching
nkern
committed
c2a50c8
added churn parameters for FM
nkern
committed
8542f9f
scale model for VE infernece
nkern
committed
409bb98
modified: transform.py
nkern
committed
e77d006
updated readme
nkern
committed
392fb00