學習率
- 查看當前學習率參數值:
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之學習率衰減