在调试pytorch程序时,当我们使用4个GPU时,传入模型的forward的批大小为原始批大小的1/4
pytorch调试的坑--多GPU调试时forward的批大小为总批大小的1/n
最新推荐文章于 2021-09-12 10:12:16 发布
在调试pytorch程序时,当我们使用4个GPU时,传入模型的forward的批大小为原始批大小的1/4