rwightman HF staff commited on
Commit
0714ea7
1 Parent(s): 3648cdd

Update model config and README

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -5,11 +5,11 @@ tags:
5
  library_name: timm
6
  license: apache-2.0
7
  datasets:
8
- - imagenet-1k
9
  ---
10
  # Model card for tf_efficientnetv2_b3.in21k
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
@@ -21,7 +21,7 @@ A EfficientNet-v2 image classification model. Trained on ImageNet-1k in Tensorfl
21
  - Image size: train = 240 x 240, test = 300 x 300
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
 
5
  library_name: timm
6
  license: apache-2.0
7
  datasets:
8
+ - imagenet-21k
9
  ---
10
  # Model card for tf_efficientnetv2_b3.in21k
11
 
12
+ A EfficientNet-v2 image classification model. Trained on ImageNet-21k in Tensorflow by paper authors, ported to PyTorch by Ross Wightman.
13
 
14
 
15
  ## Model Details
 
21
  - Image size: train = 240 x 240, test = 300 x 300
22
  - **Papers:**
23
  - EfficientNetV2: Smaller Models and Faster Training: https://arxiv.org/abs/2104.00298
24
+ - **Dataset:** ImageNet-21k
25
  - **Original:** https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet
26
 
27
  ## Model Usage