add attribute

This commit is contained in:
Li Wenyun 2023-12-03 11:34:01 +08:00
parent 3cc4dcf1d8
commit 9cb2d6bbf8
1 changed files with 5 additions and 0 deletions

View File

@ -13,6 +13,11 @@ paths:
classifier: checkpoint/
stylegan: pretrained_models/stylegan2-ffhq-config-f.pt
prompt: red lipstick
# available attributes
# ['Blond_Hair', 'Wavy_Hair', 'Young', 'Eyeglasses', 'Heavy_Makeup', 'Rosy_Cheeks',
# 'Chubby', 'Mouth_Slightly_Open', 'Bushy_Eyebrows', 'Wearing_Lipstick', 'Smiling',
# 'Arched_Eyebrows', 'Bangs', 'Wearing_Earrings', 'Bags_Under_Eyes', 'Receding_Hairline', 'Pale_Skin']
optim:
batch_size: 8