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

update git ignore

This commit is contained in:
leewlving 2023-12-06 17:36:23 +08:00
parent 2bb39f4970
commit 790867dd9f
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
.gitignore vendored
View File

@ -209,4 +209,7 @@ pyrightconfig.json
# End of https://www.toptal.com/developers/gitignore/api/linux,python,visualstudiocode # End of https://www.toptal.com/developers/gitignore/api/linux,python,visualstudiocode
*.pt *.pt
*.pth
dataset/
outputs/

9
config/config.yaml
View File

@ -3,15 +3,16 @@ classifier:
model: resnet18 model: resnet18
lr: 0.01 lr: 0.01
momentum: 0.9 momentum: 0.9
num_epochs: 200 num_epochs: 25
num_workers : 4 num_workers : 4
batch_size: 64
paths: paths:
gender_dataset: ./CelebA_HQ_face_gender_dataset gender_dataset: ./dataset/CelebA_HQ_face_gender_dataset
identity_dataset: ./CelebA_HQ_face_identity_dataset identity_dataset: ./dataset/CelebA_HQ_facial_identity_dataset
inverter_cfg: secret inverter_cfg: secret
classifier: checkpoint/ classifier: checkpoint
stylegan: checkpoint/stylegan2-ffhq-config-f.pt stylegan: checkpoint/stylegan2-ffhq-config-f.pt
adv_embedding: pretrained_models adv_embedding: pretrained_models