{ | |
"G_kwargs": { | |
"class_name": "training.networks_stylegan2.Generator", | |
"z_dim": 512, | |
"w_dim": 512, | |
"mapping_kwargs": { | |
"num_layers": 8 | |
}, | |
"channel_base": 32768, | |
"channel_max": 512, | |
"fused_modconv_default": "inference_only" | |
}, | |
"D_kwargs": { | |
"class_name": "training.networks_stylegan2.Discriminator", | |
"block_kwargs": { | |
"freeze_layers": 0 | |
}, | |
"mapping_kwargs": {}, | |
"epilogue_kwargs": { | |
"mbstd_group_size": 4 | |
}, | |
"channel_base": 32768, | |
"channel_max": 512 | |
}, | |
"G_opt_kwargs": { | |
"class_name": "torch.optim.Adam", | |
"betas": [ | |
0, | |
0.99 | |
], | |
"eps": 1e-08, | |
"lr": 0.002 | |
}, | |
"D_opt_kwargs": { | |
"class_name": "torch.optim.Adam", | |
"betas": [ | |
0, | |
0.99 | |
], | |
"eps": 1e-08, | |
"lr": 0.002 | |
}, | |
"loss_kwargs": { | |
"class_name": "training.loss.StyleGAN2Loss", | |
"r1_gamma": 8.2, | |
"style_mixing_prob": 0.9, | |
"pl_weight": 2, | |
"pl_no_weight_grad": true, | |
"blur_init_sigma": 0 | |
}, | |
"data_loader_kwargs": { | |
"pin_memory": true, | |
"prefetch_factor": 2, | |
"num_workers": 3 | |
}, | |
"training_set_kwargs": { | |
"class_name": "training.dataset.ImageFolderDataset", | |
"path": "/net/cephfs/shares/weddigen.ki.phf.uzh/common/OMNIART/dataset_stylegan_pytorch", | |
"use_labels": false, | |
"max_size": 1187043, | |
"xflip": false, | |
"resolution": 256, | |
"random_seed": 0 | |
}, | |
"num_gpus": 1, | |
"batch_size": 32, | |
"batch_gpu": 32, | |
"metrics": [ | |
"fid50k_full" | |
], | |
"total_kimg": 15000, | |
"kimg_per_tick": 4, | |
"image_snapshot_ticks": 50, | |
"network_snapshot_ticks": 50, | |
"random_seed": 0, | |
"ema_kimg": 10.0, | |
"G_reg_interval": 4, | |
"augment_kwargs": { | |
"class_name": "training.augment.AugmentPipe", | |
"xflip": 1, | |
"rotate90": 1, | |
"xint": 1, | |
"scale": 1, | |
"rotate": 1, | |
"aniso": 1, | |
"xfrac": 1, | |
"brightness": 1, | |
"contrast": 1, | |
"lumaflip": 1, | |
"hue": 1, | |
"saturation": 1 | |
}, | |
"ada_target": 0.6, | |
"resume_pkl": "/home/ludosc/data/train_outputs/network-snapshot-005000.pkl", | |
"ada_kimg": 100, | |
"ema_rampup": null, | |
"run_dir": "/home/ludosc/data/train_outputs/00040-stylegan2-dataset_stylegan_pytorch-gpus1-batch32-gamma8.2" | |
} |