PRadecki commited on
Commit
eb910bb
·
verified ·
1 Parent(s): d335554

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -10
README.md CHANGED
@@ -6,23 +6,21 @@ pipeline_tag: image-classification
6
  library_name: timm
7
  license: apache-2.0
8
  ---
9
- # Model card for convnextv2-tiny-k5-in1k-128-sup
10
 
11
- ConvNeXtV2-Tiny trained from scratch on ImageNet-1K @128x128 (k=5) using the ConvNeXt (V1) supervised recipe (Table 5; 300ep, AdamW, cosine, warmup 20, mixup 0.8, cutmix 1.0, reprob 0.25, EMA 0.9999). Trained with timm/train.py bf16.
 
 
 
 
12
 
13
  ## Model Details
14
- - **Run Dir:** /mnt/data/training-runs/convnextv2_tiny-128-k5-sup-e300-bf16-gb4096
15
- - **Experiment:** convnextv2_tiny-128-k5-sup-e300-bf16-gb4096
16
  - **Model:** convnextv2_tiny
17
  - **Model Kwargs:**
18
  - kernel_sizes: 5
19
  - **Input Size:**
20
- - 3
21
- - 128
22
- - 128
23
- - **Checkpoint:** model_best.pth.tar
24
- - **Weights Key:** state_dict_ema
25
- - **Best Metric:** 79.73000010971069
26
 
27
  ## Model Usage
28
  ```python
 
6
  library_name: timm
7
  license: apache-2.0
8
  ---
9
+ # Model card for convnextv2-tiny-k5-in1k-128
10
 
11
+ ConvNeXtV2-Tiny (k=5) trained from scratch on ImageNet-1K @128x128.
12
+
13
+ ## Training Recipe:
14
+ - ConvNeXt supervised recipe -- Table 5.
15
+ - BF16 AMP
16
 
17
  ## Model Details
 
 
18
  - **Model:** convnextv2_tiny
19
  - **Model Kwargs:**
20
  - kernel_sizes: 5
21
  - **Input Size:**
22
+ - 3x128x128
23
+ - **Top-1 Accuracy:** 79.73000010971069
 
 
 
 
24
 
25
  ## Model Usage
26
  ```python