果然1060不配跑万能模型么

mingyan

星级打分

• 1
• 2
• 3
• 4
• 5

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

我1060 6g的显卡，下了吴彦祖的刘亦菲万能模型。想着训练看看，结果参数配置到是否将模型及优化器放到gpu这里就报错了。求教大佬们这啥意思是不是显卡带不起？

markedge

mingyan 发表于 2021-5-28 23:58
bs已经改到2了还是报错，不能再小了

resolution: 256
ae_dims: 352
e_dims: 88
  d_dims: 88  

都设的太高了
这四项都已经决定显存要求不会太低的了
Gan_power开了也很吃显存


6G一般默认最好的上限是
resolution:224
ae_dims: 256
e_dims: 64
  d_dims: 64

至于更极端的我还没试过
你可以试试把优化器优先改成N 会调用一些共享显存（内存）来用  不过好像不会调用全部共享显存 但是这已经是不够显存下加显存的方案了
BS你先尝试最低看看


如果实在不行学我自己训练一个模型出来把。。

markedge

把连接发过来   或者把模型参数发过来看看

whl716694

我1070也跑不动

yws233

不是直接跑 参数需要改的吧

mingyan

markedge 发表于 2021-5-28 16:15
把连接发过来   或者把模型参数发过来看看

==------------- Model Options -------------==
==                                         ==
==            resolution: 256              ==
==             face_type: f                ==
==     models_opt_on_gpu: True             ==
==                 archi: df-ud            ==
==               ae_dims: 352              ==
==                e_dims: 88               ==
==                d_dims: 88               ==
==           d_mask_dims: 28               ==
==       masked_training: True             ==
==           uniform_yaw: False            ==
==            lr_dropout: n                ==
==           random_warp: False            ==
==             gan_power: 0.01             ==
==       true_face_power: 0.01             ==
==      face_style_power: 0.0              ==
==        bg_style_power: 0.0              ==
==               ct_mode: rct              ==
==              clipgrad: True             ==
==              pretrain: False            ==
==       autobackup_hour: 12               ==
== write_preview_history: True             ==
==           target_iter: 0                ==
==           random_flip: False            ==
==            batch_size: 8                ==
==       eyes_mouth_prio: True             ==
==             adabelief: True             ==
==        gan_patch_size: 32               ==
==              gan_dims: 16               ==

cyjcyj

我1060也不行

hainaby

把BS改小应该就行

mingyan

hainaby 发表于 2021-5-28 23:56
把BS改小应该就行

bs已经改到2了还是报错，不能再小了

sdsi

用192的丹，不要下载参数太高的，下载下面参数的模型BS4肯定能跑：
resolution: 256
ae_dims: 64
e_dims: 64
d_dims: 22

我的750Ti 4G都能跑起来