Upload README.md with huggingface_hub
Browse files
README.md
ADDED
|
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training Logs - EfficientNet-B3
|
| 2 |
+
## Experiment: 20251110_181006
|
| 3 |
+
|
| 4 |
+
### Model Architecture
|
| 5 |
+
- **Base Model**: EfficientNet-B3 (pretrained on ImageNet)
|
| 6 |
+
- **Task**: Multi-task learning (Age Regression + Gender Classification)
|
| 7 |
+
- **Input Size**: 256x256
|
| 8 |
+
|
| 9 |
+
### Training Configuration
|
| 10 |
+
- **Batch Size**: 16
|
| 11 |
+
- **Learning Rate**: 0.0002
|
| 12 |
+
- **Optimizer**: AdamW (weight_decay=1e-4)
|
| 13 |
+
- **Scheduler**: CosineAnnealingWarmRestarts
|
| 14 |
+
- **Epochs Trained**: 12 / 40
|
| 15 |
+
|
| 16 |
+
### Best Results
|
| 17 |
+
- **Validation Harmonic Mean**: 0.9683
|
| 18 |
+
- **Validation Age RMSE**: 0.1185 years
|
| 19 |
+
- **Validation Gender F1**: 0.9439
|
| 20 |
+
- **Best Epoch**: 8
|
| 21 |
+
|
| 22 |
+
### Final Metrics (Last Epoch)
|
| 23 |
+
- **Val Loss**: 0.2825
|
| 24 |
+
- **Val Harmonic Mean**: 0.9680
|
| 25 |
+
- **Val Age RMSE**: 0.1185
|
| 26 |
+
- **Val Age MAE**: 0.0850
|
| 27 |
+
- **Val Gender F1**: 0.9432
|
| 28 |
+
|
| 29 |
+
### Files in this Dataset
|
| 30 |
+
- `pytorch_lightning/metrics_20251110_181006.csv` - Complete training history
|
| 31 |
+
- `summaries/training_summary_20251110_181006.json` - Detailed metrics and config
|
| 32 |
+
- `trackio/trackio_logs_20251110_181006.zip` - TrackIO logs (if available)
|
| 33 |
+
|
| 34 |
+
### Next Steps
|
| 35 |
+
Use the best checkpoint for inference:
|
| 36 |
+
```python
|
| 37 |
+
checkpoint_path = "checkpoints/efficientnet/effnet-epoch=08-val_harmonic_mean=0.9683.ckpt"
|
| 38 |
+
```
|