pytorch lightning 使用小结(占坑)

  • 2021 年 4 月 9 日
  • AI

嗯,lightning真香,lightning的封装恰到好处,不像skorch和huggingface 的trainer那样太过完整,也不像原生torch那样那么麻烦,动不动百行代码,可贵的是,其设计的哲学很棒:

灵活自定义的部分基本保留了torch自身的语法,torch用户无缝衔接,学习成本低;

一些基本不怎么改动的地方,例如gpu的配置,amp的精度设置,早停,便捷的评估指标甚至是tpu的各种繁琐的设定等等则进行了较为完整的封装。

希望以后能多一点这样类型的工作,这才是我觉得最完美的工具,在灵活性和便利性上进行了perfect的折衷。

占坑,有空再写;