时间换显存，小显存福音，2G显存也能练论坛中猫之汉化丹

nepgear · 发表于 2022-8-10 15:09:11

星级打分

1
2
3
4
5

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

如题，我的电脑显存只有2G，之前一直用的kaggle，被封之后一直没得用。论坛中的打着小显存的其他丹也试了，失败了很多。甚至RG版也不行。
我的解决方法：修改参数[y] 将模型和优化器放在GPU上 ( y/n ?:help ) : n
训练会慢很多，但至少从不能用变成能用了。
========================模型概要========================

               模型名字: 猫之汉化神丹224 df-d_SAEHD

               当前迭代: 978850

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

         resolution: 224
         face_type: wf
   models_opt_on_gpu: False
               archi: df-d
            ae_dims: 258
            e_dims: 66
            d_dims: 64
         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: False
   random_hsv_power: 0.0
   true_face_power: 0.0
   face_style_power: 0.0
      bg_style_power: 0.0
            ct_mode: none
            clipgrad: True
            pretrain: False
   autobackup_hour: 0
write_preview_history: False
         target_iter: 0
   random_src_flip: False
   random_dst_flip: True
         batch_size: 2
         gan_power: 0.0
      gan_patch_size: 24
            gan_dims: 16
               猫の汉化: 淘宝地址:t.hk.uy/4ks
               商业合作: 商业定制视频、出售高阶模型
               联系方式: QQ微信：564646676
               此丹免费: 严禁转卖
               警惕骗子: 盗卖死妈
            官方中文论坛: dfldata.cc
retraining_samples: True

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

               设备编号: 0
               设备名称: NVIDIA GeForce GTX 1050
               显存大小: 1.31GB