显存显示与实际不符的问题

雁回翔 · 发表于 2022-2-20 12:05:23

星级打分

1
2
3
4
5

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

===============================模型概要===============================

               模型名字: WJ2_SAEHD

               当前迭代: 258351

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

         resolution: 128
         face_type: wf
   models_opt_on_gpu: True
               archi: liae-ud
            ae_dims: 128
            e_dims: 32
            d_dims: 32
         d_mask_dims: 16
   masked_training: True
   eyes_mouth_prio: False
         uniform_yaw: False
      blur_out_mask: False
         adabelief: True
         lr_dropout: n
         random_warp: True
   random_hsv_power: 0.0
   true_face_power: 0.0
   face_style_power: 0.0
      bg_style_power: 0.0
            ct_mode: rct
            clipgrad: True
            pretrain: False
   autobackup_hour: 3
write_preview_history: False
         target_iter: 0
   random_src_flip: False
   random_dst_flip: False
         batch_size: 16
         gan_power: 0.0
      gan_patch_size: 12
            gan_dims: 16

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

               设备编号: 0
               设备名称: NVIDIA GeForce RTX 3050 Ti Laptop GPU
               显存大小: 1.62GB

==============================================================

实际是4G显存，但是总是显示1.62G，上次有位兄台说要调大batch_size，我试了从4调到16还是没有用，不知道这个情况对训练有没有什么影响，如果有影响要怎么办才能调到和实际显存差不多的大小。