Feat: training experiments#29

Merged

anna-grim merged 56 commits into

AllenNeuralDynamics:mainfrom

carshadi:feat-training-experiments

Jul 16, 2026

Commits on Jul 8, 2026

feat: add dynamic-range-preserving intensity transforms

carshadi
and
claude
committed
fix: symmetric skeleton sampling, validation sigma, device threading

carshadi
and
claude
committed
feat: wire intensity transform through train/inference, drop clip

carshadi
and
claude
committed
feat: neurite-preservation metrics and joint checkpoint selection

carshadi
and
claude
committed
feat: foreground-preserving target and signal-preserving loss

carshadi
and
claude
committed
feat: signal-balanced patch sampling for thin fibers

carshadi
and
claude
committed
feat: residual denoising output and GroupNorm in the 3D U-Net

carshadi
and
claude
committed
feat: per-brain background offset wiring

carshadi
and
claude
committed
feat: per-volume/per-brain offset in the offline evaluators

carshadi
and
claude
committed
fix: strip gs:// prefix from segmentation kvstore path

carshadi
and
claude
committed

Commits on Jul 9, 2026

Commits on Jul 10, 2026

Commits on Jul 11, 2026

Commits on Jul 13, 2026

Train on the full patch cache per epoch
carshadi
committed

Commits on Jul 14, 2026

Update train_bm4dnet.py
carshadi
committed

Commits on Jul 15, 2026

Add option to disable AMP during training/validation
carshadi
committed