RaghavIITM commited on
Commit
da5dcec
·
verified ·
1 Parent(s): 63a9cb6

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +38 -0
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
+ ```