def configure_optimizers( self, ) -> Tuple[List[torch.optim.Optimizer], List]: optim = d2_build_optimizer(self.cfg, self.model) lr_scheduler = d2_build_lr_scheduler(self.cfg, optim) return [optim], [{"scheduler": lr_scheduler, "interval": "step"}]
def build_lr_scheduler(self, cfg, optimizer): return d2_build_lr_scheduler(cfg, optimizer)