pytorch訓練時accuracy收斂緩慢的可能原因

今天在使用pytorch搭建LSTM網絡跑THUC數據集時,發現每遍歷一遍50000條數據,準確率只增長了0.001-0.01。後來發現是學習率設置小了。 提高學習率後,模型在前幾個epoch就快速收斂了。 這種小錯誤以後要避免呀!=_=
相關文章
相關標籤/搜索