update

train_yamls/train_512FM.yaml

@@ -50,7 +50,7 @@ d_optim_config:
   eps: !!float 1e-8
 
 id_weight: 20.0
-reconstruct_weight: 1.0
+reconstruct_weight: 10.0
 feature_match_weight: 10.0
 
 # Log 

train_yamls/train_512FM_Modulation.yaml

@@ -0,0 +1,63 @@
+# Related scripts
+train_script_name: FM
+
+# models' scripts
+model_configs:
+  g_model:
+    script: Generator_reduce
+    class_name: Generator
+    module_params:
+      g_conv_dim: 512
+      g_kernel_size: 3
+      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: 12
+
+# Dataset
+dataloader: VGGFace2HQ
+dataset_name: vggface2_hq
+dataset_params:
+  random_seed: 1234
+  dataloader_workers: 8
+
+eval_dataloader: DIV2K_hdf5
+eval_dataset_name: DF2K_H5_Eval
+eval_batch_size: 2
+
+# Dataset
+
+# Optimizer
+optim_type: Adam
+g_optim_config:
+  lr:  0.0004
+  betas: [ 0, 0.99]
+  eps: !!float 1e-8
+
+d_optim_config:
+  lr:  0.0004
+  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

train_yamls/train_GramFM.yaml

@@ -0,0 +1,63 @@
+# Related scripts
+train_script_name: GramFM
+
+# models' scripts
+model_configs:
+  g_model:
+    script: Generator
+    class_name: Generator
+    module_params:
+      g_conv_dim: 512
+      g_kernel_size: 3
+      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: 12
+
+# Dataset
+dataloader: VGGFace2HQ
+dataset_name: vggface2_hq
+dataset_params:
+  random_seed: 1234
+  dataloader_workers: 8
+
+eval_dataloader: DIV2K_hdf5
+eval_dataset_name: DF2K_H5_Eval
+eval_batch_size: 2
+
+# Dataset
+
+# Optimizer
+optim_type: Adam
+g_optim_config:
+  lr:  0.0004
+  betas: [ 0, 0.99]
+  eps: !!float 1e-8
+
+d_optim_config:
+  lr:  0.0004
+  betas: [ 0, 0.99]
+  eps: !!float 1e-8
+
+id_weight: 20.0
+reconstruct_weight: 10.0
+feature_match_weight: 100.0
+
+# Log 
+log_step: 300
+model_save_step: 10000
+total_step: 1000000
+sample_step: 1000
+checkpoint_names:
+  generator_name: Generator
+  discriminator_name: Discriminator