| { | |
| "model_name": "CIFAR-10 Diffusion Model", | |
| "architecture": "SimpleUNet", | |
| "dataset": "CIFAR-10", | |
| "training_details": { | |
| "epochs": 20, | |
| "batch_size": 128, | |
| "learning_rate": 0.0001, | |
| "optimizer": "AdamW", | |
| "scheduler": "CosineAnnealingLR", | |
| "parameters": 16808835, | |
| "training_time_minutes": 14.54, | |
| "final_loss": 0.0363, | |
| "best_loss": 0.0358 | |
| }, | |
| "model_config": { | |
| "in_channels": 3, | |
| "out_channels": 3, | |
| "time_emb_dim": 128, | |
| "image_size": 32 | |
| }, | |
| "diffusion_config": { | |
| "num_timesteps": 1000, | |
| "beta_start": 0.0001, | |
| "beta_end": 0.02, | |
| "schedule": "linear" | |
| }, | |
| "hardware": { | |
| "gpu": "NVIDIA GeForce RTX 3060", | |
| "vram_used": "0.43 GB", | |
| "total_vram": "11.66 GB" | |
| }, | |
| "created_date": "2025-07-19T17:59:48.665409", | |
| "framework": "PyTorch", | |
| "python_version": "3.12" | |
| } |