| { | |
| "num_gpus": 2, | |
| "image_snapshot_ticks": 20, | |
| "network_snapshot_ticks": 20, | |
| "metrics": [], | |
| "random_seed": 0, | |
| "slideflow_kwargs": { | |
| "project_path": "/mnt/data/projects/TCGA_THCA_BRAF", | |
| "tile_px": 512, | |
| "tile_um": 302, | |
| "model_type": "categorical", | |
| "outcome_label_headers": [ | |
| "brs_class" | |
| ], | |
| "outcome_labels": { | |
| "0": "Braf-like", | |
| "1": "Ras-like" | |
| }, | |
| "filters": { | |
| "brs_class": [ | |
| "Braf-like", | |
| "Ras-like" | |
| ] | |
| } | |
| }, | |
| "training_set_kwargs": { | |
| "class_name": "training.slideflow_dataset.SlideflowIterator", | |
| "path": "/mnt/data/gan/slideflow_config.json", | |
| "use_labels": true, | |
| "max_size": null, | |
| "xflip": true, | |
| "resolution": 512 | |
| }, | |
| "data_loader_kwargs": { | |
| "pin_memory": true, | |
| "num_workers": 3, | |
| "prefetch_factor": 2 | |
| }, | |
| "G_kwargs": { | |
| "class_name": "training.networks.Generator", | |
| "z_dim": 512, | |
| "w_dim": 512, | |
| "interp_embed": false, | |
| "mapping_kwargs": { | |
| "num_layers": 2 | |
| }, | |
| "synthesis_kwargs": { | |
| "channel_base": 32768, | |
| "channel_max": 512, | |
| "num_fp16_res": 4, | |
| "conv_clamp": 256 | |
| } | |
| }, | |
| "D_kwargs": { | |
| "class_name": "training.networks.Discriminator", | |
| "block_kwargs": {}, | |
| "interp_embed": false, | |
| "mapping_kwargs": {}, | |
| "epilogue_kwargs": { | |
| "mbstd_group_size": 4 | |
| }, | |
| "channel_base": 32768, | |
| "channel_max": 512, | |
| "num_fp16_res": 4, | |
| "conv_clamp": 256 | |
| }, | |
| "G_opt_kwargs": { | |
| "class_name": "torch.optim.Adam", | |
| "lr": 0.0025, | |
| "betas": [ | |
| 0, | |
| 0.99 | |
| ], | |
| "eps": 1e-08 | |
| }, | |
| "D_opt_kwargs": { | |
| "class_name": "torch.optim.Adam", | |
| "lr": 0.0025, | |
| "betas": [ | |
| 0, | |
| 0.99 | |
| ], | |
| "eps": 1e-08 | |
| }, | |
| "loss_kwargs": { | |
| "class_name": "training.loss.StyleGAN2Loss", | |
| "r1_gamma": 3.2768 | |
| }, | |
| "total_kimg": 25000, | |
| "batch_size": 16, | |
| "batch_gpu": 8, | |
| "ema_kimg": 5.0, | |
| "ema_rampup": 0.05, | |
| "ada_target": 0.6, | |
| "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, | |
| "imgfilter": 1, | |
| "noise": 1, | |
| "cutout": 1 | |
| }, | |
| "allow_tf32": true, | |
| "run_dir": "/mnt/data/gan/brs-new-training/00012-slideflow-test-cond-mirror-auto2-bgcfnc" | |
| } |