Add training summary
Browse files
README.md
ADDED
|
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# resnet50
|
| 2 |
+
|
| 3 |
+
**Best Validation Accuracy:** `0.9050`
|
| 4 |
+
|
| 5 |
+
## Metadata
|
| 6 |
+
- **Model Name**: `resnet50`
|
| 7 |
+
- **Optimizer**: `adamw`
|
| 8 |
+
- **Scheduler**: `cosine`
|
| 9 |
+
- **Weight Decay**: `0.0005`
|
| 10 |
+
- **Warmup Epochs**: `3`
|
| 11 |
+
- **Patience**: `7`
|
| 12 |
+
- **Amp**: `True`
|
| 13 |
+
- **Seed**: `42`
|
| 14 |
+
- **Batch Size**: `32`
|
| 15 |
+
- **Initial Lr**: `0.0005`
|
| 16 |
+
- **Total Epochs Ran**: `14`
|
| 17 |
+
- **Early Stopped**: `True`
|
| 18 |
+
- **Training Time Seconds**: `33306.75762796402`
|
| 19 |
+
- **Num Parameters**: `22067009`
|
| 20 |
+
- **Device**: `NVIDIA A100-SXM4-40GB`
|
| 21 |
+
- **Run Id**: `resnet-tuned`
|
| 22 |
+
|
| 23 |
+
## Training Configuration
|
| 24 |
+
- Epochs: `14`
|
| 25 |
+
- Batch size: `32`
|
| 26 |
+
- Learning rate (initial): `0.0005`
|
| 27 |
+
|
| 28 |
+
## Training Logs (Per Epoch)
|
| 29 |
+
| Epoch | Train Loss | Train Acc | Val Loss | Val Acc | LR |
|
| 30 |
+
|-------|------------|-----------|----------|---------|----|
|
| 31 |
+
| 1 | 0.2662 | 0.8921 | 0.2935 | 0.8786 | 0.000167 |
|
| 32 |
+
| 2 | 0.2245 | 0.9125 | 0.3381 | 0.8763 | 0.000333 |
|
| 33 |
+
| 3 | 0.2090 | 0.9196 | 0.4355 | 0.8174 | 0.000500 |
|
| 34 |
+
| 4 | 0.1841 | 0.9304 | 0.3426 | 0.8751 | 0.000500 |
|
| 35 |
+
| 5 | 0.1678 | 0.9376 | 0.3754 | 0.8611 | 0.000499 |
|
| 36 |
+
| 6 | 0.1554 | 0.9430 | 0.4509 | 0.8415 | 0.000495 |
|
| 37 |
+
| 7 | 0.1447 | 0.9468 | 0.2905 | 0.9050 | 0.000488 |
|
| 38 |
+
| 8 | 0.1373 | 0.9502 | 0.4505 | 0.8444 | 0.000478 |
|
| 39 |
+
| 9 | 0.1297 | 0.9539 | 0.4726 | 0.8473 | 0.000467 |
|
| 40 |
+
| 10 | 0.1224 | 0.9569 | 0.3371 | 0.8867 | 0.000452 |
|
| 41 |
+
| 11 | 0.1167 | 0.9583 | 0.3782 | 0.8741 | 0.000436 |
|
| 42 |
+
| 12 | 0.1115 | 0.9604 | 0.4738 | 0.8645 | 0.000417 |
|
| 43 |
+
| 13 | 0.1052 | 0.9626 | 0.3477 | 0.8896 | 0.000397 |
|
| 44 |
+
| 14 | 0.0998 | 0.9647 | 0.4067 | 0.8845 | 0.000375 |
|