2022年3月30日
pytorch学习笔记——训练时显存逐渐增加，几个epoch后out-of-memory
摘要： 问题起因：笔者想把别人的torch的代码复制到笔者的代码框架下，从而引起的显存爆炸问题 该bug在困扰了笔者三天的情况下，和学长一同解决了该bug，故在此记录这次艰辛的debug之路。 尝试思路1：检查是否存在保留loss的情况下是否使用了 item() 取值，经检查，并没有 尝试思路2：按照网上的 阅读全文
posted @ 2022-03-30 23:21 ISGuXing 阅读(5934) 评论(0) 推荐(1) 编辑