20 lines
380 B
YAML
20 lines
380 B
YAML
dataset: gender_dataset
|
|
classifier:
|
|
model: resnet18
|
|
lr: 0.01
|
|
momentum: 0.9
|
|
num_epochs: 200
|
|
|
|
|
|
paths:
|
|
gender_dataset: ./CelebA_HQ_face_gender_dataset
|
|
identity_dataset: ./CelebA_HQ_face_identity_dataset
|
|
inverter_cfg: secret
|
|
classifier: checkpoint/
|
|
stylegan: pretrained_models/stylegan2-ffhq-config-f.pt
|
|
|
|
|
|
optim:
|
|
batch_size: 8
|
|
num_epochs: 200
|
|
num_workers : 4 |