deepfold.train.lr_scheduler.AlphaFoldLRScheduler

class deepfold.train.lr_scheduler.AlphaFoldLRScheduler(init_lr: float, final_lr: float, warmup_lr_length: int, init_lr_length: int, optimizer: Optimizer)

AlphaFold learning rate schedule.

Suppl. ‘1.11.3 Optimization details’.

__init__(init_lr: float, final_lr: float, warmup_lr_length: int, init_lr_length: int, optimizer: Optimizer) → None

Methods

__init__(init_lr, final_lr, ...)
step(iteration)