| { | |
| "model_type": "dcgan", | |
| "architecture": { | |
| "generator": { | |
| "latent_size": 100, | |
| "feature_maps": 64, | |
| "output_channels": 3, | |
| "image_size": 64 | |
| }, | |
| "discriminator": { | |
| "feature_maps": 64, | |
| "input_channels": 3, | |
| "image_size": 64 | |
| } | |
| }, | |
| "training": { | |
| "epochs": 100, | |
| "learning_rate": 0.0002, | |
| "beta1": 0.5, | |
| "batch_size": 32, | |
| "mixed_precision": true, | |
| "loss_function": "BCEWithLogitsLoss" | |
| }, | |
| "dataset": { | |
| "image_size": 64, | |
| "normalization_mean": [0.5, 0.5, 0.5], | |
| "normalization_std": [0.5, 0.5, 0.5] | |
| }, | |
| "generation": { | |
| "default_seed": null, | |
| "default_images": 16 | |
| }, | |
| "framework": { | |
| "library": "pytorch", | |
| "torch_version": "2.x" | |
| } | |
| } |