Hugging Face's logo

ivensamdh
/
gender

Image Classification
Transformers
PyTorch
vit
Generated from Trainer
Inference Endpoints
Model card Files Community
2
ivensamdh commited on
Commit
76ef15f
•
1 Parent(s): 570bc49

Model save

Browse files
Files changed (3) hide show
  1. config.json +4 -4
  2. pytorch_model.bin +2 -2
  3. training_args.bin +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "facebook/deit-tiny-patch16-224",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
@@ -7,13 +7,13 @@
7
  "encoder_stride": 16,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
- "hidden_size": 192,
11
  "image_size": 224,
12
  "initializer_range": 0.02,
13
- "intermediate_size": 768,
14
  "layer_norm_eps": 1e-12,
15
  "model_type": "vit",
16
- "num_attention_heads": 3,
17
  "num_channels": 3,
18
  "num_hidden_layers": 12,
19
  "patch_size": 16,
 
1
  {
2
+ "_name_or_path": "facebook/deit-small-patch16-224",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
7
  "encoder_stride": 16,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 384,
11
  "image_size": 224,
12
  "initializer_range": 0.02,
13
+ "intermediate_size": 1536,
14
  "layer_norm_eps": 1e-12,
15
  "model_type": "vit",
16
+ "num_attention_heads": 6,
17
  "num_channels": 3,
18
  "num_hidden_layers": 12,
19
  "patch_size": 16,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a219180cb398eaadcb52d2b5242ec58f26b63d1ca29a7b7b1443fc1983ad5f18
3
- size 22166637
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:132ef324b75d3e4b0feb0a126ec11abe99c249ff2b539b95e5829d087e3b5602
3
+ size 86733677
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b99e7a07d739f1acc0f222e773a038ca7f79a87bf7385b94f19b53aacb9806a5
3
  size 3515
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bb70ceb3dd4a9486c3d944eea7a75b9a9176c013862dc2c385e7211aa1c00536
3
  size 3515