distillation

2022-03-04 19:04:36 +08:00
parent 39964bf613
commit e02f756116
10 changed files with 1232 additions and 23 deletions
@@ -0,0 +1,64 @@
+# Related scripts
+train_script_name: multi_gpu
+
+# models' scripts
+model_configs:
+  g_model:
+    script: Generator_ori_modulation_config
+    class_name: Generator
+    module_params:
+      id_dim: 512
+      g_kernel_size: 3
+      in_channel: 8
+      res_num: 9
+
+  d_model:
+    script: projected_discriminator
+    class_name: ProjectedDiscriminator
+    module_params:
+      diffaug: False
+      interp224: False
+      backbone_kwargs: {}
+
+arcface_ckpt: arcface_ckpt/arcface_checkpoint.tar
+
+# Training information
+batch_size: 32
+
+# Dataset
+dataloader: VGGFace2HQ_multigpu
+dataset_name: vggface2_hq
+dataset_params:
+  random_seed: 1234
+  dataloader_workers: 4
+
+eval_dataloader: DIV2K_hdf5
+eval_dataset_name: DF2K_H5_Eval
+eval_batch_size: 2
+
+# Dataset
+
+# Optimizer
+optim_type: Adam
+g_optim_config:
+  lr:  0.0006
+  betas: [ 0, 0.99]
+  eps: !!float 1e-8
+
+d_optim_config:
+  lr:  0.0006
+  betas: [ 0, 0.99]
+  eps: !!float 1e-8
+
+id_weight: 20.0
+reconstruct_weight: 10.0
+feature_match_weight: 10.0
+
+# Log 
+log_step: 300
+model_save_step: 10000
+sample_step: 1000
+total_step: 1000000
+checkpoint_names:
+  generator_name: Generator
+  discriminator_name: Discriminator