refactoring noise_schedule and time schedule into base class#1736

Merged

manuelgloeckler merged 38 commits intomainfrom

psteinb-explicit_noise_schedules-1437

Feb 4, 2026

Commits on Mar 20, 2025

Commits on Jan 22, 2026

Commits on Jan 23, 2026

WIP : use time schedule in loss function, address device issues
Camille Touron
committed

Commits on Jan 26, 2026

call solve_schedule in validation step
Camille Touron
committed
add solve_schedule method, call train_schedule in loss
Camille Touron
committed
call the solve schedule during sampling with SDE
Camille Touron
committed

Commits on Jan 27, 2026

add a solve_schedule function in the conditional vf estimator class to unify training in vftrainer class
Camille Touron
committed

Commits on Jan 29, 2026

make the solve schedule deterministic
Camille Touron
committed
corrections on solve schedule
Camille Touron
committed
WIP : create solve schedule in base class
Camille Touron
committed

Commits on Jan 30, 2026

modify arguments of solve schedule
Camille Touron
committed
change train_schedule + docstrings fixes + device handling
Camille Touron
committed
include validation times nugget to avoid instabilities during training
Camille Touron
committed
change the nb of simulations for ve option
Camille Touron
committed
change device in solve schedule
Camille Touron
committed
add noise schedule in VE subclass
Camille Touron
committed
reshape noise schedule output in VE class
Camille Touron
committed
reshape noise schedule output in VE class
Camille Touron
committed

Commits on Jan 31, 2026

add tests on train and solve schedule shapes, devices, bounds
Camille Touron
committed

Commits on Feb 2, 2026

formatting and changing tests
Camille Touron
committed