不懂就问，我这情况是换卡还是怎么操作？

dd373664

星级打分

• 1
• 2
• 3
• 4
• 5

平均分:NAN  参与人数:0  我的评分:未评

现在用的是RTX4060显卡，训练调试选项如图，批量2都跑不起来，一直提示


Error: 2 root error(s) found.
  (0) Resource exhausted: OOM when allocating tensor with shape[3,3,2048,2048] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
         [[node src_dst_opt_1/Select_26 (defined at C:\Users\1\软件\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\core\leras\ops\__init__.py:212) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info. This isn't available when running in Eager mode.

这个要换一张16G的显卡？还是可以调一下显存开始跑？图3的设置批量2可以跑，但改一下设置之后就跑不动了。
来个大佬解答一下

llon

这个丹是自己新建的吗？我看论坛分辨率224的丹auto encode一般是256，你这个ae_mids=512，是很高的参数，16G显存也许还跑不动

醉卧二两仙

批量数据超了显存，可以尝试RG版本，时间换显存，或者直接4090 走起

dd373664

llon 发表于 2023-9-27 15:57
这个丹是自己新建的吗？我看论坛分辨率224的丹auto encode一般是256，你这个ae_mids=512，是很高的参数，16 ...

对，自己弄的丹，src最初16w张，被我删到6w，我看教程他们15g的显存也是调512。跟着调的

dd373664

醉卧二两仙 发表于 2023-9-27 16:03
批量数据超了显存，可以尝试RG版本，时间换显存，或者直接4090 走起

RG版本是什么？这个ae_mids=512这里太高了？

醉卧二两仙

这个数值确实大了点，猫224神丹是这个一半。还有你有两个 power 0.1    1.0 ，一般模型不开的.论坛里搜搜 ICE 木兰  版本，批量应该能开到4

dd373664

醉卧二两仙 发表于 2023-9-27 16:18
这个数值确实大了点，猫224神丹是这个一半。还有你有两个 power 0.1    1.0 ，一般模型不开的.论坛里搜搜 I ...

嗯，那如果我把这个调成246-256，再换一张16G显卡，是不是可以跑批量8？

llon

dd373664 发表于 2023-9-27 16:31
嗯，那如果我把这个调成246-256，再换一张16G显卡，是不是可以跑批量8？

改256，bs=8，4060不用换

pestxo

有详细教程么？

番茄哥

显存溢出