pytorch lightning 使用小結(占坑)
- 2021 年 4 月 9 日
- AI
嗯,lightning真香,lightning的封裝恰到好處,不像skorch和huggingface 的trainer那樣太過完整,也不像原生torch那樣那麼麻煩,動不動百行代碼,可貴的是,其設計的哲學很棒:
靈活自定義的部分基本保留了torch自身的語法,torch用戶無縫銜接,學習成本低;
一些基本不怎麼改動的地方,例如gpu的配置,amp的精度設置,早停,便捷的評估指標甚至是tpu的各種繁瑣的設定等等則進行了較為完整的封裝。
希望以後能多一點這樣類型的工作,這才是我覺得最完美的工具,在靈活性和便利性上進行了perfect的折衷。
占坑,有空再寫;