flow_matching.path.scheduler
#
Scheduler#
Base Scheduler class. |
|
CondOT Scheduler. |
|
Cosine Scheduler. |
|
Variance Preserving Scheduler. |
|
Polynomial Scheduler. |
ScheduleTransformedModel#
ScheduleTransformedModel wraps a given model and converts its scheduler
Change of scheduler for a velocity model. |