Downloads · 30 days
0
age-gender-efficientnet
RaghavIITM/age-gender-efficientnet
age-gender-efficientnet is a machine learning model from RaghavIITM. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
- Base Model: EfficientNet-B3 (pretrained on ImageNet) - Task: Multi-task learning (Age Regression + Gender Classification) - Input Size: 256x256
Access
Public
Updated Nov 11, 2025
Repo size
624 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.ckpt936 MB · 100%
At a glance
- Access
- Public
- Created
- Nov 10, 2025
- Updated
- Nov 11, 2025
- SHA
- 78cc2d1e
Model card
From the Hugging Face model README
Training Logs - EfficientNet-B3
Experiment: 20251111_031508
Model Architecture
- Base Model: EfficientNet-B3 (pretrained on ImageNet)
- Task: Multi-task learning (Age Regression + Gender Classification)
- Input Size: 256x256
Training Configuration
- Batch Size: 16
- Learning Rate: 0.0002
- Optimizer: AdamW (weight_decay=1e-4)
- Scheduler: CosineAnnealingWarmRestarts
- Epochs Trained: 34 / 40
Best Results
- Validation Harmonic Mean: 0.9699
- Validation Age RMSE: 0.1175 years
- Validation Gender F1: 0.9469
- Best Epoch: 12
Final Metrics (Last Epoch)
- Val Loss: 0.2847
- Val Harmonic Mean: 0.9650
- Val Age RMSE: 0.1177
- Val Age MAE: 0.0875
- Val Gender F1: 0.9375
Files in this Dataset
pytorch_lightning/metrics_20251111_031508.csv- Complete training historysummaries/training_summary_20251111_031508.json- Detailed metrics and configtrackio/trackio_logs_20251111_031508.zip- TrackIO logs (if available)
Next Steps
Use the best checkpoint for inference:
checkpoint_path = "checkpoints/efficientnet/effnet-epoch=12-val_harmonic_mean=0.9699.ckpt"
- Created
- Nov 10, 2025
- Updated
- Nov 11, 2025