timm
/

Image Classification
timm
PyTorch
Safetensors
rwightman HF staff commited on
Commit
613f470
1 Parent(s): 9a2df5e

Update model config and README

Browse files
Files changed (1) hide show
  1. README.md +3 -1
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