agile_transfer_metface-oil1024x1024.py

encoder_ckpt_path = 'work_dirs/pre-trained/agile_encoder_ffhq1024x1024_lr_1e-4_500kiter_20211201_112111-fb1312dc.pth'  # noqa

stylegan_weights = 'work_dirs/pre-trained/agile_transfer_metfaces-oil1024x1024_zplus_lpips0.5_freezeD5_ada_bs4x2_lr_1e-4_1600iter_20211104_134350-2b99cb9b.pth'  # noqa

model = dict(type='PSPEncoderDecoder',
             encoder=dict(type='VAEStyleEncoder',
                          num_layers=50,
                          pretrained=dict(ckpt_path=encoder_ckpt_path,
                                          prefix='encoder',
                                          strict=False)),
             decoder=dict(type='SwapStyleGANv2Generator',
                          out_size=1024,
                          style_channels=512,
                          num_mlps=8,
                          pretrained=dict(ckpt_path=stylegan_weights,
                                          prefix='generator_ema')),
             pool_size=(1024, 1024),
             id_lambda=0.0,
             lpips_lambda=0.0,
             id_ckpt=None,
             kl_loss=None,
             train_cfg=None,
             test_cfg=None)

train_cfg = None
test_cfg = None