liwenyun
/
GanAttack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

更新 config/config.yaml

This commit is contained in:
liwenyun 2023-12-12 16:41:13 +08:00
parent 08eee04d1e
commit a358d00102
1 changed files with 33 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
config/config.yaml
View File

@ -1,32 +1,34 @@
dataset: identity_dataset dataset: identity_dataset
classifier: classifier:
model: resnet18 model: resnet18
lr: 0.01 lr: 0.01
momentum: 0.9 momentum: 0.9
num_epochs: 120 num_epochs: 120
num_workers : 4 num_workers : 4
batch_size: 128 batch_size: 128
paths: paths:
gender_dataset: ./dataset/CelebA_HQ_face_gender_dataset gender_dataset: ./dataset/CelebA_HQ_face_gender_dataset
identity_dataset: dataset/CelebA_HQ_facial_identity_dataset identity_dataset: dataset/CelebA_HQ_facial_identity_dataset
inverter_cfg: /dataset/face_identity/psp_ffhq_encode.pt base_code: dataset/CelebA_HQ_facial_identity_dataset
classifier: checkpoint detail_code: dataset/CelebA_HQ_facial_identity_dataset
stylegan: checkpoint/stylegan2-ffhq-config-f.pt inverter_cfg: /dataset/face_identity/psp_ffhq_encode.pt
adv_embedding: pretrained_models classifier: checkpoint
stylegan: checkpoint/stylegan2-ffhq-config-f.pt
prompt: red lipstick adv_embedding: pretrained_models
# available attributes
# ['Blond_Hair', 'Wavy_Hair', 'Young', 'Eyeglasses', 'Heavy_Makeup', 'Rosy_Cheeks', prompt: red lipstick
# 'Chubby', 'Mouth_Slightly_Open', 'Bushy_Eyebrows', 'Wearing_Lipstick', 'Smiling', # available attributes
# 'Arched_Eyebrows', 'Bangs', 'Wearing_Earrings', 'Bags_Under_Eyes', 'Receding_Hairline', 'Pale_Skin'] # ['Blond_Hair', 'Wavy_Hair', 'Young', 'Eyeglasses', 'Heavy_Makeup', 'Rosy_Cheeks',
# 'Chubby', 'Mouth_Slightly_Open', 'Bushy_Eyebrows', 'Wearing_Lipstick', 'Smiling',
optim: # 'Arched_Eyebrows', 'Bangs', 'Wearing_Earrings', 'Bags_Under_Eyes', 'Receding_Hairline', 'Pale_Skin']
batch_size: 1
num_epochs: 200 optim:
num_workers : 4 batch_size: 1
images_resize: 256 num_epochs: 200
alpha: 0.1 num_workers : 4
beta: 1 images_resize: 256
alpha: 0.1
beta: 1
delta: 1 delta: 1