| { | |
| "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" | |
| } |