请问如何修改学习率衰减器

[blue247973104]

您代码中的学习率衰减器为
CFG.TRAIN.LR_SCHEDULER.TYPE = "MultiStepLR"
CFG.TRAIN.LR_SCHEDULER.PARAM = {
"milestones": [30, 40, 45],
"gamma": 0.1
}
请问如何更改成
torch.optim.lr_scheduler.ReduceLROnPlateau(optimizer, mode='min', factor=0.1, patience=10,
verbose=False, threshold=0.001, threshold_mode='rel', cooldown=0, min_lr=2e-6, eps=1e-08)这种
或者别的学习率衰减器

[zezhishao]

BasicTS目前还不支持ReduceLROnPlateau，其他的学习率衰减器都是支持的。

你可以直接修改CFG.TRAIN.LR_SCHEDULER.TYPE为其名字，然后将衰减器的参数放在这个字典里CFG.TRAIN.LR_SCHEDULER.TYPE。
例如：

CFG.TRAIN.LR_SCHEDULER = EasyDict()
CFG.TRAIN.LR_SCHEDULER.TYPE = "ExponentialLR"
CFG.TRAIN.LR_SCHEDULER.PARAM = {
    "gamma" : 1,
    "verbose" : False
}

只要是继承自torch.optim.lr_scheduler._LRScheduler的BasicTS都支持，ReduceLROnPlateau继承自object，目前还不兼容。