随笔分类 -  神经网络训练

减小运行内存的方法
摘要：1、 写成函数的形式，通过调用函数，返回想要的结果，中间变量在返回后会清空 函数返回值越少越好，_ 虽然不会显示变量，但会占内存 2、清除中间变量函数 if hasattr(torch.cuda, 'empty_cache'): torch.cuda.empty_cache() 3、设置变量初始值， 阅读全文
posted @ 2022-10-06 21:27 乙二 阅读(125) 评论(0) 推荐(0)