怎麼把預訓的Model變成萬用Model？

main9dog

星级打分

• 1
• 2
• 3
• 4
• 5

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

最近抓了火炬版，作者有放個連結有 190萬訓練的Model，但這個好像是什麼「預訓」Model


pretrain: True


要怎麼把這個Model轉成萬用Model拿來用，我還在用很久以前版上有給我的192萬用Model。
想用這個224的看看，不知道怎麼改成不是「預訓」



================== Model Summary ===================
==                                                ==
==            Model name: new_SAEHD               ==
==                                                ==
==     Current iteration: 1916330                 ==
==                                                ==
==---------------- Model Options -----------------==
==                                                ==
==            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: False                   ==
==           uniform_yaw: True                    ==
==         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: True                    ==
==       autobackup_hour: 0                       ==
== write_preview_history: False                   ==
==           target_iter: 0                       ==
==       random_src_flip: False                   ==
==       random_dst_flip: False                   ==
==            batch_size: 8                       ==
==             gan_power: 0.0                     ==
==        gan_patch_size: 28                      ==
==              gan_dims: 16                      ==
==                                                ==
==------------------ Running On ------------------==
==                                                ==
==          Device index: 0                       ==
==                  Name: NVIDIA GeForce RTX 3090 ==
==                  VRAM: 21.15GB                 ==
==                                                ==
====================================================