pytorch 中ReduceLROnPlateau 的使用
看代码提示里写的 ‘min’是判断 schedular.step(val_loss) 里输入的val_loss是否减小来更改lr的值的我是这样写的:scheduler = torch.optim.lr_scheduler.ReduceLROnPlateau(optimizer, mode='min', factor=0.5, patience=2, verbose=True, threshold=0.0001, threshold_mode='rel', cooldown=0, min_..
原创
2021-10-04 22:41:31 ·
5591 阅读 ·
1 评论