iamcode6 commited on
Commit
a5c2a62
·
verified ·
1 Parent(s): ef68231

Upload folder using huggingface_hub

Browse files
Files changed (5) hide show
  1. best.pt +3 -0
  2. classification_report.txt +11 -0
  3. config.yaml +54 -0
  4. confusion_matrix.csv +5 -0
  5. metrics.json +12 -0
best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e369616d25d4620d5389fdd7163100b46f01d6ad02fdd507985f15059c5f34d
3
+ size 1216359894
classification_report.txt ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ precision recall f1-score support
2
+
3
+ carmenta 0.8077 0.7500 0.7778 28
4
+ healthy 0.9728 0.9670 0.9699 333
5
+ moniliasis 0.9057 0.8727 0.8889 55
6
+ phytophthora 0.7797 0.8519 0.8142 54
7
+ witches_broom 0.8000 0.8889 0.8421 9
8
+
9
+ accuracy 0.9290 479
10
+ macro avg 0.8532 0.8661 0.8586 479
11
+ weighted avg 0.9304 0.9290 0.9294 479
config.yaml ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Cocoa pod 5-class — EVA-02-Large fine-tuning on MI300X
2
+ # Backbone: EVA-02-L ViT-L/14 pretrained MIM-CLIP, fine-tuned IN22k+IN1k @ 448
3
+ # Dataset : merged LatAm + Peru YOLO crops (~4.9k crops, 5 classes)
4
+ #
5
+ # Resolution is 448 — quadruples per-image memory vs Track 2's 224×224 DINOv2,
6
+ # so batch_size and num_workers are scaled down. Saturate later if VRAM allows.
7
+
8
+ seed: 123
9
+
10
+ data:
11
+ # splits.json built by prepare_cocoa_data.py — paths inside it are absolute
12
+ splits_file: splits.json
13
+ num_workers: 4
14
+ pin_memory: true
15
+
16
+ model:
17
+ # timm name for EVA-02-Large pretrained MIM (LAION) + IN22k + IN1k FT @ 448
18
+ name: eva02_large_patch14_448.mim_m38m_ft_in22k_in1k
19
+ num_classes: 5
20
+ drop_path_rate: 0.1
21
+ img_size: 448
22
+
23
+ train:
24
+ # Single-phase fine-tune (matches the simplified track2/train.py behavior)
25
+ lr: 1.0e-4
26
+ epochs: 20 # ~5k crops × 20 epochs ≈ 100k steps total at BS=64
27
+ batch_size: 64 # safe at 448px on a 192GB MI300X; raise if you have headroom
28
+ grad_accum_steps: 1
29
+ weight_decay: 0.05
30
+ betas: [0.9, 0.999]
31
+ label_smoothing: 0.1
32
+ mixup_alpha: 0.1 # matches the reduced setting that worked in Track 2
33
+ cutmix_alpha: 0.5
34
+ mixup_prob: 0.5
35
+ grad_clip: 1.0
36
+ amp_dtype: bfloat16 # MI300X has native bf16
37
+ compile: false # enable after the first stable run if you want speed
38
+ grad_checkpointing: true # 448px attn maps are big; checkpoint to free VRAM
39
+
40
+ augment:
41
+ rand_augment_n: 2
42
+ rand_augment_m: 9
43
+ random_erasing_p: 0.25
44
+ horizontal_flip: true
45
+ vertical_flip: true # cocoa pods don't have a strong up/down orientation
46
+
47
+ eval:
48
+ batch_size: 64
49
+ tta_rounds: 10
50
+
51
+ log:
52
+ every_n_steps: 50
53
+ checkpoint_dir: runs
54
+ save_best_metric: val_macro_f1 # rare classes (witches_broom, carmenta) → optimize macro F1
confusion_matrix.csv ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ 21,4,0,3,0
2
+ 3,322,2,4,2
3
+ 0,1,48,6,0
4
+ 1,4,3,46,0
5
+ 1,0,0,0,8
metrics.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "standard": {
3
+ "acc": 0.9206680584551148,
4
+ "macro_f1": 0.8458037279646249
5
+ },
6
+ "tta": {
7
+ "acc": 0.9290187891440501,
8
+ "macro_f1": 0.8585621479864496,
9
+ "rounds": 10
10
+ },
11
+ "improvement_pp": 0.8350730688935326
12
+ }