请问遮罩训练XSeg) train报错是什么原因？

20210901 · 发表于 2021-9-6 02:05:21

滚石发表于 2021-9-5 13:44
OOM是显存不足
xseg训练很吃显存的，低配显卡跑不动

谢谢提醒，我鼓捣了半天，用CPU可以跑遮罩了，但速度非常慢
降低设置后用显卡也可以跑十几秒，然后又报同样OOM错，看来就是显存问题了。想不到炼遮罩居然比直接炼丹更难。

我是3060TI显卡，8G显存，内存条32G，共享给了GPU 16G
但是显卡的共享GPU并不会工作，请问有什么办法设置DFL让跑遮罩可以利用共享GPU吗？

共享内存.png

20210901 · 发表于 2021-9-6 02:08:14

haha007 发表于 2021-9-5 11:40
系統問題！1050都能跑10了，更何況3060

直接练模型没问题，就是遮罩不行，请问你的1050能炼遮罩么？和src，dst的图片大小数量有关系么？

dkrhfk · 发表于 2021-9-6 12:36:24

系統問題！1050都能跑10了，更何況3060

Victor_lu · 发表于 2021-9-6 18:37:35

就离谱，我BS开4，运行不了。开2，还是不行。现在试着增大BS，结果bs开到8，就能用了。看看等会儿会不会崩吧。

hwbeva · 发表于 2021-9-6 20:05:35

感谢分享

20210901 · 发表于 2021-9-7 04:06:32

Victor_lu 发表于 2021-9-6 18:37
就离谱，我BS开4，运行不了。开2，还是不行。现在试着增大BS，结果bs开到8，就能用了。看看等会儿会不会崩 ...

多谢老兄指点！

我开到8也能跑了。。开低了反而不行。。。。太离谱了

www8333888 · 发表于 2021-9-7 15:03:09

不懂帮顶

		自动登录	找回密码
密码			立即注册（仅限QQ邮箱）