ViT-LoRA-Food101-0310 / run_summary.json
nanchennn's picture
Upload folder using huggingface_hub
b564748 verified
Raw
History Blame Contribute Delete
2.27 kB
{
"run_id": "ViT-LoRA-Food101-0310",
"base_model": "google/vit-large-patch16-224-in21k",
"dataset": "food101",
"num_total_classes": 101,
"num_selected_classes": 50,
"selected_class_ids": [
1,
6,
7,
8,
9,
10,
12,
14,
17,
18,
23,
25,
30,
31,
32,
33,
36,
40,
44,
48,
50,
53,
55,
56,
60,
61,
62,
64,
65,
67,
69,
74,
75,
76,
80,
81,
83,
84,
86,
87,
88,
90,
91,
92,
93,
95,
96,
97,
98,
99
],
"selected_class_names": [
"baby_back_ribs",
"beignets",
"bibimbap",
"bread_pudding",
"breakfast_burrito",
"bruschetta",
"cannoli",
"carrot_cake",
"cheese_plate",
"chicken_curry",
"churros",
"club_sandwich",
"deviled_eggs",
"donuts",
"dumplings",
"edamame",
"falafel",
"french_fries",
"fried_rice",
"greek_salad",
"grilled_salmon",
"hamburger",
"hot_dog",
"huevos_rancheros",
"lobster_bisque",
"lobster_roll_sandwich",
"macaroni_and_cheese",
"miso_soup",
"mussels",
"omelette",
"oysters",
"peking_duck",
"pho",
"pizza",
"pulled_pork_sandwich",
"ramen",
"red_velvet_cake",
"risotto",
"sashimi",
"scallops",
"seaweed_salad",
"spaghetti_bolognese",
"spaghetti_carbonara",
"spring_rolls",
"steak",
"sushi",
"tacos",
"takoyaki",
"tiramisu",
"tuna_tartare"
],
"seed": 328518879,
"lora": {
"rank": 32,
"target_modules": [
"query",
"key",
"value"
]
},
"hparams": {
"learning_rate": 5e-05,
"weight_decay": 0.005,
"lr_scheduler_raw": "constant",
"epochs": 6,
"random_crop": false,
"random_flip": false,
"global_batch_size": 64,
"per_device_batch_size": 64,
"grad_accumulation_steps": 1
},
"sizes": {
"train_examples": 37500,
"eval_examples": 12500
},
"metrics": {
"top1_accuracy": 0.94504,
"eval_loss": 0.20931996405124664,
"eval_accuracy": 0.94504,
"eval_runtime": 24.9369,
"eval_samples_per_second": 501.266,
"eval_steps_per_second": 7.86,
"epoch": 6.0
}
}