CalorieCLIP / config.json
Jared
v1.1: Improved model trained on 13K samples - MAE 51.4 cal (was 58.3)
f8a9cee
{
"model_type": "calorie-clip",
"version": "1.0.0",
"base_model": "ViT-B-32",
"pretrained": "openai",
"hidden_dim": 512,
"output_dim": 1,
"task": "calorie_regression",
"training": {
"epochs": 30,
"batch_size": 16,
"learning_rate_clip": 1e-5,
"learning_rate_head": 1e-3,
"loss_function": "huber",
"optimizer": "adamw",
"fine_tuned_layers": "last_2_transformer_blocks"
},
"performance": {
"mae": 51.4,
"within_50_cal": 67.6,
"within_100_cal": 90.5,
"test_samples": 1951,
"training_samples": 11053
},
"preprocessing": {
"image_size": 224,
"mean": [0.48145466, 0.4578275, 0.40821073],
"std": [0.26862954, 0.26130258, 0.27577711]
},
"license": "MIT",
"author": "Haplo LLC",
"intended_use": "Food calorie estimation from images"
}