torchcell.losses.mle_wasserstein.AdaptiveWeighting

class AdaptiveWeighting(warmup_epochs: int = 100, stable_epoch: int = 500)

Bases: object

Manages adaptive weighting for buffer contributions during training.

get_buffer_weight(epoch: int) → float

Get buffer weight based on training epoch.