pytorch减少显存占用和加速(checkpoint(会慢),apex(会块))
1、pytorch 的 checkpoint 是一个可以用时间换空间的技术,很多情况下可以轻松实现 batch_size 翻倍的效果.快速使用链接:https://blog.csdn.net/ONE_SIX_MIX/article/details/896393342、apex 混合精度训练快速使用链接:https://zhuanlan.zhihu.com/p/79887894...