Python d2_build_lr_scheduler示例

示例#1

    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"}]

示例#2

文件： default_runner.py 项目： ananthsub/d2go

 def build_lr_scheduler(self, cfg, optimizer):
     return d2_build_lr_scheduler(cfg, optimizer)