(optimizer, init_lr, epoch, decay_rate=0.1, decay_epoch=30)
| 46 | |
| 47 | |
| 48 | def adjust_lr(optimizer, init_lr, epoch, decay_rate=0.1, decay_epoch=30): |
| 49 | decay = decay_rate ** (epoch // decay_epoch) |
| 50 | for param_group in optimizer.param_groups: |
| 51 | param_group['lr'] = decay*init_lr |
| 52 | lr=param_group['lr'] |
| 53 | return lr |
| 54 | |
| 55 | # def adjust_lr(optimizer, init_lr, epoch, decay_rate=0.1, decay_epoch=30): |
| 56 | # decay = decay_rate ** (epoch // decay_epoch) |
nothing calls this directly
no outgoing calls
no test coverage detected