This commit is contained in:
chenxuanhong
2022-03-03 21:19:52 +08:00
parent 860713323d
commit 39964bf613
12 changed files with 6121 additions and 41 deletions
+2 -1
View File
@@ -13,6 +13,7 @@ model_configs:
res_num: 9
up_mode: bilinear
aggregator: "invo"
res_mode: "invo"
d_model:
script: projected_discriminator
@@ -25,7 +26,7 @@ model_configs:
arcface_ckpt: arcface_ckpt/arcface_checkpoint.tar
# Training information
batch_size: 24
batch_size: 64
# Dataset
dataloader: VGGFace2HQ_multigpu