nathanReitinger commited on
Commit
17a31f3
·
verified ·
1 Parent(s): 35f3dc5

Training in progress, epoch 1

Browse files
README.md CHANGED
@@ -1,8 +1,6 @@
1
  ---
2
  license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
4
- datasets:
5
- - fashion_mnist
6
  tags:
7
  - generated_from_trainer
8
  metrics:
 
1
  ---
2
  license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
 
 
4
  tags:
5
  - generated_from_trainer
6
  metrics:
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "google/vit-base-patch16-224-in21k",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
@@ -7,7 +7,7 @@
7
  "encoder_stride": 16,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
- "hidden_size": 768,
11
  "id2label": {
12
  "0": "T - shirt / top",
13
  "1": "Trouser",
@@ -22,7 +22,7 @@
22
  },
23
  "image_size": 224,
24
  "initializer_range": 0.02,
25
- "intermediate_size": 3072,
26
  "label2id": {
27
  "Ankle boot": "9",
28
  "Bag": "8",
@@ -37,10 +37,10 @@
37
  },
38
  "layer_norm_eps": 1e-12,
39
  "model_type": "vit",
40
- "num_attention_heads": 12,
41
  "num_channels": 3,
42
- "num_hidden_layers": 12,
43
- "patch_size": 16,
44
  "problem_type": "single_label_classification",
45
  "qkv_bias": true,
46
  "torch_dtype": "float32",
 
1
  {
2
+ "_name_or_path": "google/vit-huge-patch14-224-in21k",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
7
  "encoder_stride": 16,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 1280,
11
  "id2label": {
12
  "0": "T - shirt / top",
13
  "1": "Trouser",
 
22
  },
23
  "image_size": 224,
24
  "initializer_range": 0.02,
25
+ "intermediate_size": 5120,
26
  "label2id": {
27
  "Ankle boot": "9",
28
  "Bag": "8",
 
37
  },
38
  "layer_norm_eps": 1e-12,
39
  "model_type": "vit",
40
+ "num_attention_heads": 16,
41
  "num_channels": 3,
42
+ "num_hidden_layers": 32,
43
+ "patch_size": 14,
44
  "problem_type": "single_label_classification",
45
  "qkv_bias": true,
46
  "torch_dtype": "float32",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:57e9084e99876f056bd5184d3972e09b1e630476ec3cdd7958e60c28567f85e5
3
- size 343248584
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b80334d35d46b5acd623708ec4c577c24b9fdca1d95502f0e847556c413e452
3
+ size 2523172784
runs/May01_22-24-25_tron05.umiacs.umd.edu/events.out.tfevents.1714616666.tron05.umiacs.umd.edu.1876157.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e1365814b64683320798062c59456c56ae8b2d7d81e2dca7362bada6bae36eb
3
+ size 6702
runs/May01_22-31-23_tron05.umiacs.umd.edu/events.out.tfevents.1714617084.tron05.umiacs.umd.edu.1876721.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2ea9a3defe6509c788fde954448b7d71da0af42d0bedd178f376a0e29d86a7e
3
+ size 21145
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c943166912b7acf7036f17232b694ae0786edb03c6fe1677fced461872b599c6
3
  size 5048
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc85443ee15c9b4db51fd2654dd7e41bfa4b08601e19fe5bd82816c50a4ee1e4
3
  size 5048