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
password: secret
optim:
batch_size: 8
num_workers : 4