XHChen0528
70 lines
1.3 KiB
YAML
70 lines
1.3 KiB
YAML
# 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 |