File size: 843 Bytes
da54531 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
{
"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"
} |