deepfold.utils.swa_utils

Classes

AlphaFoldSWA(alphafold, enabled, decay_rate)	Weight averaging wrapper.
swa_avg_fn(decay_rate)	Averaging function for EMA with configurable decay rate.