| """ | |
| Copied from RT-DETR (https://github.com/lyuwenyu/RT-DETR) | |
| Copyright(c) 2023 lyuwenyu. All Rights Reserved. | |
| """ | |
| import torch.optim as optim | |
| import torch.optim.lr_scheduler as lr_scheduler | |
| from ..core import register | |
| __all__ = ["AdamW", "SGD", "Adam", "MultiStepLR", "CosineAnnealingLR", "OneCycleLR", "LambdaLR"] | |
| SGD = register()(optim.SGD) | |
| Adam = register()(optim.Adam) | |
| AdamW = register()(optim.AdamW) | |
| MultiStepLR = register()(lr_scheduler.MultiStepLR) | |
| CosineAnnealingLR = register()(lr_scheduler.CosineAnnealingLR) | |
| OneCycleLR = register()(lr_scheduler.OneCycleLR) | |
| LambdaLR = register()(lr_scheduler.LambdaLR) | |