一开learning rate drop就崩是怎么回事？

benlei

星级打分

• 1
• 2
• 3
• 4
• 5

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

为什么一开learning rate drop这个参数就崩啊？？？
================== Model Summary ===================
==                                                ==
==            Model name: GXT256DFUD_SAEHD        ==
==                                                ==
==     Current iteration: 264462                  ==
==                                                ==
==---------------- Model Options -----------------==
==                                                ==
==            resolution: 256                     ==
==             face_type: wf                      ==
==     models_opt_on_gpu: True                    ==
==                 archi: df-ud                   ==
==               ae_dims: 256                     ==
==                e_dims: 64                      ==
==                d_dims: 64                      ==
==           d_mask_dims: 22                      ==
==       masked_training: True                    ==
==           uniform_yaw: False                   ==
==            lr_dropout: n                       ==
==           random_warp: True                    ==
==             gan_power: 0.0                     ==
==       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_flip: True                    ==
==            batch_size: 8                       ==
==       eyes_mouth_prio: True                    ==
==         blur_out_mask: False                   ==
==             adabelief: True                    ==
==      random_hsv_power: 0.0                     ==
==       random_src_flip: False                   ==
==       random_dst_flip: True                    ==
==        gan_patch_size: 32                      ==
==              gan_dims: 16                      ==
==          QQ_XXXXXXXXXX: Speechless              ==
==                                                ==
==------------------ Running On ------------------==
==                                                ==
==          Device index: 0                       ==
==                  Name: NVIDIA GeForce RTX 3070 ==
==                  VRAM: 6.84GB                  ==
==                                                ==
====================================================

山野村夫

batch_size调4

wcw888

闪退？

lwz123

显存太小了

denvor

models_opt_on_gpu: True
改成Falase
不把models 放在显存里，能节约一部分显存，稍微慢一点点。
不然就调低 batchsize

longsxp

开这个吃显存，调小一点BS试试

LAON

显存太小了

benlei

denvor 发表于 2022-3-22 11:03
models_opt_on_gpu: True
改成Falase
不把models 放在显存里，能节约一部分显存，稍微慢一点点。

batchsize开4也会报错。

benlei

山野村夫 发表于 2022-3-22 10:33
batch_size调4

开4也要报错

lucq3000

benlei 发表于 2022-3-22 12:32
开4也要报错

那就改成2