# Related scripts
train_script_name: multi_gpu_cycle

# models' scripts
model_configs:
  g_model:
    script: Generator_LSTU_config
    class_name: Generator
    module_params:
      id_dim: 512
      g_kernel_size: 3
      in_channel: 64
      res_num: 9
      up_mode: bilinear
      aggregator: "conv"
      res_mode: "conv"

  d_model:
    script: projected_discriminator
    class_name: ProjectedDiscriminator
    module_params:
      diffaug: False
      interp224: False
      backbone_kwargs: {}

# arcface_ckpt: arcface_torch/checkpoints/glint360k_cosface_r100_fp16_backbone.pth
arcface_ckpt: arcface_ckpt/arcface_checkpoint.tar

# Training information
batch_size: 20

# Dataset
dataloader: VGGFace2HQ_multigpu
dataset_name: vggface2_hq
dataset_params:
  random_seed: 1234
  dataloader_workers: 6

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
rec_feature_match_weight: 10.0
cycle_feature_match_weight: 10.0
cycle_weight: 10.0

# Log 
log_step: 400
model_save_step: 10000
total_step: 1000000
sample_step: 1000
checkpoint_names:
  generator_name: Generator
  discriminator_name: Discriminator