|
{ |
|
"experiment_name": "cc_icgan_biggan_imagenet_res256", |
|
"run_setup": "local_debug", |
|
"deterministic_run": true, |
|
"num_workers": 10, |
|
|
|
"ddp_train": true, |
|
"n_nodes": 1, |
|
"n_gpus_per_node": 8, |
|
"hflips": true, |
|
"DA": true, |
|
"DiffAugment": "translation", |
|
|
|
"test_every": 10, |
|
"save_every": 1, |
|
"num_epochs": 3000, |
|
"es_patience": 50, |
|
"shuffle": true, |
|
|
|
"G_eval_mode": true, |
|
"ema": true, |
|
"use_ema": true, |
|
"num_G_accumulations": 1, |
|
"num_D_accumulations": 1, |
|
"num_D_steps": 2, |
|
|
|
"class_cond": true, |
|
"instance_cond": true, |
|
"which_knn_balance": "instance_balance", |
|
"hier": true, |
|
"resolution": 256, |
|
"G_attn": "64", |
|
"D_attn": "64", |
|
"shared_dim": 128, |
|
"shared_dim_feat": 512, |
|
"G_shared": true, |
|
"G_shared_feat": true, |
|
|
|
"k_nn": 5, |
|
"feature_extractor": "classification", |
|
|
|
"batch_size": 16, |
|
"D_lr": 1e-4, |
|
"G_lr": 1e-4, |
|
"G_ch": 64, |
|
"D_ch": 64, |
|
|
|
"longtail": true, |
|
"longtail_gen": true, |
|
"use_balanced_sampler": false, |
|
"custom_distrib_gen": false, |
|
"longtail_temperature": 1, |
|
|
|
"load_weights": "" |
|
|
|
} |