8G的3070显卡BIZE开1还提示OMM

as3231177

星级打分

• 1
• 2
• 3
• 4
• 5

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

用的是坛主的LIAE迪丽热巴神丹，不知道该如何优化设置才能让我进行下去
可不能浪费那大几百灵石呀！！


----------------------模型选项----------------------

            resolution: 224
             face_type: wf
     models_opt_on_gpu: True
                 archi: liae-udt
               ae_dims: 512
                e_dims: 64
                d_dims: 64
           d_mask_dims: 32
       masked_training: True
       eyes_mouth_prio: True
           uniform_yaw: True
         blur_out_mask: True
             adabelief: False
            lr_dropout: y
           random_warp: True
      random_hsv_power: 0.1
       true_face_power: 0.0
      face_style_power: 0.0
        bg_style_power: 0.0
               ct_mode: none
              clipgrad: False
              pretrain: False
       autobackup_hour: 0
write_preview_history: False
           target_iter: 0
       random_src_flip: False
       random_dst_flip: True
            batch_size: 1
             gan_power: 0.0
        gan_patch_size: 28
              gan_dims: 16

----------------------运行信息----------------------

                  设备编号: 0
                  设备名称: NVIDIA GeForce RTX 3070
                  显存大小: 5.34GB





保存时间|迭代次数|单次时间|SRC损失|DST损失
Error: OOM when allocating tensor with shape[2048,2048,3,3] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
         [[node gradients/Conv2D_40_grad/Conv2DBackpropFilter (defined at J:\NARUTO\fake\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\core\leras\ops\__init__.py:55) ]]
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.

公海Nana7mi

8g你想跑512的丹？

daiandy

人家练的就是专丹，直接合成就行，若想自己再练，把耗显存的选项该关的关，

2863748407

ae_dims: 512
你跑不了的
这个参数不能改

hideto123

3060 试过没难度

as3231177

hideto123 发表于 2022-7-12 16:42
3060 试过没难度

你的显存是多大捏，楼上有大佬说8G显存跑不了，我贼难过，浪费了呀

1234567xy

试试看RG版的，我1070TI,也是8g,跑这丹BS可以加到6，另外你系统是win11吗？？为什么可用显存才5G多

hideto123

as3231177 发表于 2022-7-12 16:50
你的显存是多大捏，楼上有大佬说8G显存跑不了，我贼难过，浪费了呀

3060 是12g

wtxx8888

LIAE-UDT资源占用不是一般的高，而且那么高的ae_dims跟d_mask_dims，你的显卡用官方版DFL肯定跑不了。

kokia033

关模型放gpu 和ada优化器 以及侧脸优化gan 随即反转 甚是  gan_patch_size:和 gan_dims:  大小 一般不能改..
减小src  与dst  aligned头像分辨像素率大小 ..  即可解决