PyTorch易混淆知識(持續更新)

學習率

  1. 查看當前學習率參數值:optimizer.param_groups[0]['lr']
  2. 訓練時更改學習率:torch.optim.lr_scheduler.StepLR(optimizer,step_size=100,gamma=0.99,last_epoch=-1)。詳解請看:TensorFlow和PyTorch之學習率衰減
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章