timm
/

Image Classification
timm
PyTorch
Safetensors
rwightman HF staff commited on
Commit
345dc60
1 Parent(s): b15933a
Files changed (3) hide show
  1. README.md +37 -0
  2. config.json +41 -0
  3. pytorch_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - image-classification
4
+ - timm
5
+ library_tag: timm
6
+ license: apache-2.0
7
+ ---
8
+ # Model card for convnext_tiny.in12k_ft_in1k
9
+
10
+ A ConvNeXt image classification model. Pretrained on ImageNet-12k (a 11821 class subset of full ImageNet-22k) and fine-tuned on ImageNet-1k by Ross Wightman.
11
+
12
+ ImageNet-12k training done on TPUs thanks to support of the [TRC](https://sites.research.google/trc/about/) program.
13
+ Fine-tuning performed on 8x GPU [Lambda Labs](https://lambdalabs.com/) cloud instances.
14
+
15
+ ## Model Details
16
+ - **Model Type:** Image classification / feature backbone
17
+
18
+ ## Citation
19
+ ```
20
+ @misc{rw2019timm,
21
+ author = {Ross Wightman},
22
+ title = {PyTorch Image Models},
23
+ year = {2019},
24
+ publisher = {GitHub},
25
+ journal = {GitHub repository},
26
+ doi = {10.5281/zenodo.4414861},
27
+ howpublished = {\url{https://github.com/rwightman/pytorch-image-models}}
28
+ }
29
+ ```
30
+ ```
31
+ @article{liu2022convnet,
32
+ author = {Zhuang Liu and Hanzi Mao and Chao-Yuan Wu and Christoph Feichtenhofer and Trevor Darrell and Saining Xie},
33
+ title = {A ConvNet for the 2020s},
34
+ journal = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
35
+ year = {2022},
36
+ }
37
+ ```
config.json ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architecture": "convnext_tiny",
3
+ "num_classes": 1000,
4
+ "num_features": 768,
5
+ "pretrained_cfg": {
6
+ "tag": "in12k_ft_in1k",
7
+ "custom_load": false,
8
+ "input_size": [
9
+ 3,
10
+ 224,
11
+ 224
12
+ ],
13
+ "test_input_size": [
14
+ 3,
15
+ 288,
16
+ 288
17
+ ],
18
+ "fixed_input_size": false,
19
+ "interpolation": "bicubic",
20
+ "crop_pct": 0.95,
21
+ "test_crop_pct": 1.0,
22
+ "crop_mode": "center",
23
+ "mean": [
24
+ 0.485,
25
+ 0.456,
26
+ 0.406
27
+ ],
28
+ "std": [
29
+ 0.229,
30
+ 0.224,
31
+ 0.225
32
+ ],
33
+ "num_classes": 1000,
34
+ "pool_size": [
35
+ 7,
36
+ 7
37
+ ],
38
+ "first_conv": "stem.0",
39
+ "classifier": "head.fc"
40
+ }
41
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1abc6ff66688819cab2f7f8b1d9e14f7830c947ad663bd3c1f6681c0d31fcce6
3
+ size 114420885