编程小白

pytorch 如何减少内存和显存

最近遇到个项目:要不以前的几个小项目 统一用一个模型来集中训练,可是我遇到了 显卡爆的局势,怎办?

  1. 老外写的提高pytorch效率的方法,包含data prefetch等
  2. Pytorch有什么节省显存的小技巧?
© 2023 编程小白. Generated with Hugo and Mainroad theme.