(此文章用于留档,方便以后遇到此问题时查找)

设置persistent_workers为True(默认False),即每个epoch后不销毁子进程,会增加内、显存占用,该参数在num_workers>0时才会生效。

Logo

免费领 200 小时云算力,进群参与显卡、AI PC 幸运抽奖

更多推荐