Search

torch.optim.lr_scheduler 종류

from torch.optim.lr_scheduler import ReduceLROnPlateau
metric이 patience epoch만큼 나아지지 않으면 factor만큼 감소 ( new_lr = lr * factor )
optimizer 에 torch.optim 객체를 넣어주면 된다
Made with