Update README.md
Browse files
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
|
| 10 |
|
| 11 |
-
ConvNeXtV2-Tiny trained from scratch on ImageNet-1K @128x128
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
-
-
|
| 21 |
-
|
| 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
|