pytorch 如何减少内存和显存

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

  1. 老外写的提高pytorch效率的方法,包含data prefetch等
  2. Pytorch有什么节省显存的小技巧?

你可能感兴趣的:(python应用,pytorch,深度学习,人工智能)