Update model config and README
Browse files
README.md
CHANGED
@@ -6,10 +6,11 @@ library_name: timm
|
|
6 |
license: apache-2.0
|
7 |
datasets:
|
8 |
- imagenet-1k
|
|
|
9 |
---
|
10 |
# Model card for tf_efficientnetv2_m.in21k_ft_in1k
|
11 |
|
12 |
-
A EfficientNet-v2 image classification model. Trained on ImageNet-1k in Tensorflow by paper authors, ported to PyTorch by Ross Wightman.
|
13 |
|
14 |
|
15 |
## Model Details
|
@@ -22,6 +23,7 @@ A EfficientNet-v2 image classification model. Trained on ImageNet-1k in Tensorfl
|
|
22 |
- **Papers:**
|
23 |
- EfficientNetV2: Smaller Models and Faster Training: https://arxiv.org/abs/2104.00298
|
24 |
- **Dataset:** ImageNet-1k
|
|
|
25 |
- **Original:** https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet
|
26 |
|
27 |
## Model Usage
|
|
|
6 |
license: apache-2.0
|
7 |
datasets:
|
8 |
- imagenet-1k
|
9 |
+
- imagenet-21k
|
10 |
---
|
11 |
# Model card for tf_efficientnetv2_m.in21k_ft_in1k
|
12 |
|
13 |
+
A EfficientNet-v2 image classification model. Trained on ImageNet-21k and fine-tuned on ImageNet-1k in Tensorflow by paper authors, ported to PyTorch by Ross Wightman.
|
14 |
|
15 |
|
16 |
## Model Details
|
|
|
23 |
- **Papers:**
|
24 |
- EfficientNetV2: Smaller Models and Faster Training: https://arxiv.org/abs/2104.00298
|
25 |
- **Dataset:** ImageNet-1k
|
26 |
+
- **Pretrain Dataset:** ImageNet-21k
|
27 |
- **Original:** https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet
|
28 |
|
29 |
## Model Usage
|