pyanomaly.core.scheduler package¶
Submodules¶
pyanomaly.core.scheduler.scheduler_api module¶
@author: Yuhao Cheng @contact: yuhao.cheng[at]outlook.com
-
class
pyanomaly.core.scheduler.scheduler_api.SchedulerAPI(cfg)¶ Bases:
object
pyanomaly.core.scheduler.schedulers module¶
-
class
pyanomaly.core.scheduler.schedulers.WarmupCosineLR(optimizer: torch.optim.optimizer.Optimizer, max_iters: int, warmup_factor: float = 0.001, warmup_iters: int = 1000, warmup_method: str = 'linear', last_epoch: int = - 1)¶ Bases:
torch.optim.lr_scheduler._LRScheduler-
get_lr() → List[float]¶
-
-
class
pyanomaly.core.scheduler.schedulers.WarmupMultiStepLR(optimizer: torch.optim.optimizer.Optimizer, milestones: List[int], gamma: float = 0.1, warmup_factor: float = 0.001, warmup_iters: int = 1000, warmup_method: str = 'linear', last_epoch: int = - 1)¶ Bases:
torch.optim.lr_scheduler._LRScheduler-
get_lr() → List[float]¶
-