logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

关于pytorch训练时GPU占用不满的问题

设置persistent_workers为True(默认False),即每个epoch后不销毁子进程,会增加内、显存占用,该参数在num_workers>0时才会生效。(此文章用于留档,方便以后遇到此问题时查找)

#pytorch#人工智能#python
到底了