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