mnist-conditional-gan / generator_training_config.json
andreribeiro87's picture
Upload generator_training_config.json with huggingface_hub
2fa83f6 verified
Raw
History Blame Contribute Delete
910 Bytes
{
"adversarial_loss": "hinge",
"batch_size": 32,
"beta1": 0.5,
"beta2": 0.999,
"checkpoint_every": 1000,
"data_cache_dir": null,
"device": "cuda",
"epochs": 25,
"eval_at_end": true,
"eval_batch_size": 32,
"eval_every": 1000,
"eval_num_fake": 2048,
"eval_num_real": 2048,
"image_size": 64,
"lambda_fm": 0.0,
"lambda_gp": 10.0,
"lambda_perc": 0.0,
"log_every": 50,
"lr_d": 0.0002,
"lr_g": 0.0002,
"max_steps": null,
"metrics_device": null,
"n_critic": 1,
"n_samples_grid": 64,
"ndf": 64,
"ngf": 64,
"num_classes": 10,
"num_workers": 2,
"nz": 100,
"output_root": "outputs",
"run_name": "gan_run",
"sample_every": 500,
"seed": 42,
"spectral_norm_d": true,
"tags": {},
"task": "mnist",
"use_bf16": true,
"use_gradient_penalty": false,
"use_wandb": true,
"wandb_entity": null,
"wandb_mode": "online",
"wandb_project": "week06-gan"
}