深度学习之性能优化之torch篇(占坑)

  • 2021 年 4 月 6 日
  • AI

最近torch玩的比较多,发现居然还有很多trick之类的可以减少内存使用率v和提高运行速度,这里开个坑,慢慢总结起来。整体来说,目前的感受就是,成熟任务用tf.keras,又好又快,一站式解决;不成熟或者是魔改用torch,半灵活性的接口设计,简单的令人发指的自定义方法以及非常完善的各类配套周边,使得torch可以轻松玩转各类新的idea。