torchcell.losses.mle_dist_supcr.AdaptiveWeighting

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

Bases: object

Manages adaptive weighting for buffer contributions during training.

get_buffer_weight(epoch: int) → float[source]

Get buffer weight based on training epoch.