一切从解决这个bug开始
改小batch size就能用了
Linux
(base) root@7f770f0ddb6c:~# nvidia-smi
我自己的经验:
10G显存跑UNET,batch size=6/8是够用的,其实batch size 对时间影响是很大的,虽然自身没有体会过,但的确经历过=6不收敛=8就有收敛趋势的情况,下图来自一个有趣的网址,解释的很好 这里
一切从解决这个bug开始
改小batch size就能用了
Linux
(base) root@7f770f0ddb6c:~# nvidia-smi
我自己的经验:
10G显存跑UNET,batch size=6/8是够用的,其实batch size 对时间影响是很大的,虽然自身没有体会过,但的确经历过=6不收敛=8就有收敛趋势的情况,下图来自一个有趣的网址,解释的很好 这里