学习率
- 查看当前学习率参数值:
optimizer.param_groups[0]['lr']
- 训练时更改学习率:
torch.optim.lr_scheduler.StepLR(optimizer,step_size=100,gamma=0.99,last_epoch=-1)
。详解请看:TensorFlow和PyTorch之学习率衰减
optimizer.param_groups[0]['lr']
torch.optim.lr_scheduler.StepLR(optimizer,step_size=100,gamma=0.99,last_epoch=-1)
。详解请看:TensorFlow和PyTorch之学习率衰减