Prahas10 commited on
Commit
3307946
·
1 Parent(s): 20f5af1

Training in progress epoch 0

Browse files
Files changed (4) hide show
  1. README.md +7 -15
  2. config.json +6 -6
  3. preprocessor_config.json +2 -2
  4. tf_model.h5 +2 -2
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  license: apache-2.0
3
- base_model: google/vit-large-patch16-224-in21k
4
  tags:
5
  - generated_from_keras_callback
6
  model-index:
@@ -13,12 +13,12 @@ probably proofread and complete it, then remove this comment. -->
13
 
14
  # Prahas10/roof-large
15
 
16
- This model is a fine-tuned version of [google/vit-large-patch16-224-in21k](https://huggingface.co/google/vit-large-patch16-224-in21k) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Train Loss: 0.1559
19
- - Validation Loss: 0.2924
20
- - Train Accuracy: 0.8909
21
- - Epoch: 8
22
 
23
  ## Model description
24
 
@@ -44,15 +44,7 @@ The following hyperparameters were used during training:
44
 
45
  | Train Loss | Validation Loss | Train Accuracy | Epoch |
46
  |:----------:|:---------------:|:--------------:|:-----:|
47
- | 1.8354 | 1.1693 | 0.6364 | 0 |
48
- | 0.6363 | 0.5030 | 0.8455 | 1 |
49
- | 0.4077 | 0.3187 | 0.9091 | 2 |
50
- | 0.2877 | 0.4201 | 0.8273 | 3 |
51
- | 0.3060 | 0.6308 | 0.8364 | 4 |
52
- | 0.3030 | 0.2840 | 0.9091 | 5 |
53
- | 0.1304 | 0.1397 | 0.9545 | 6 |
54
- | 0.3800 | 0.4405 | 0.8455 | 7 |
55
- | 0.1559 | 0.2924 | 0.8909 | 8 |
56
 
57
 
58
  ### Framework versions
 
1
  ---
2
  license: apache-2.0
3
+ base_model: google/vit-base-patch16-384
4
  tags:
5
  - generated_from_keras_callback
6
  model-index:
 
13
 
14
  # Prahas10/roof-large
15
 
16
+ This model is a fine-tuned version of [google/vit-base-patch16-384](https://huggingface.co/google/vit-base-patch16-384) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Train Loss: 1.6423
19
+ - Validation Loss: 1.0616
20
+ - Train Accuracy: 0.7091
21
+ - Epoch: 0
22
 
23
  ## Model description
24
 
 
44
 
45
  | Train Loss | Validation Loss | Train Accuracy | Epoch |
46
  |:----------:|:---------------:|:--------------:|:-----:|
47
+ | 1.6423 | 1.0616 | 0.7091 | 0 |
 
 
 
 
 
 
 
 
48
 
49
 
50
  ### Framework versions
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "google/vit-large-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": 1024,
11
  "id2label": {
12
  "0": "cteed_landmark_driftwood_shade",
13
  "1": "cteed_landmark_driftwood_sun",
@@ -25,9 +25,9 @@
25
  "8": "iko_dynasty_cornerstone_sun",
26
  "9": "iko_dynasty_granite_black_shade"
27
  },
28
- "image_size": 224,
29
  "initializer_range": 0.02,
30
- "intermediate_size": 4096,
31
  "label2id": {
32
  "cteed_landmark_driftwood_shade": "0",
33
  "cteed_landmark_driftwood_sun": "1",
@@ -47,9 +47,9 @@
47
  },
48
  "layer_norm_eps": 1e-12,
49
  "model_type": "vit",
50
- "num_attention_heads": 16,
51
  "num_channels": 3,
52
- "num_hidden_layers": 24,
53
  "patch_size": 16,
54
  "qkv_bias": true,
55
  "transformers_version": "4.38.2"
 
1
  {
2
+ "_name_or_path": "google/vit-base-patch16-384",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
7
  "encoder_stride": 16,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
  "id2label": {
12
  "0": "cteed_landmark_driftwood_shade",
13
  "1": "cteed_landmark_driftwood_sun",
 
25
  "8": "iko_dynasty_cornerstone_sun",
26
  "9": "iko_dynasty_granite_black_shade"
27
  },
28
+ "image_size": 384,
29
  "initializer_range": 0.02,
30
+ "intermediate_size": 3072,
31
  "label2id": {
32
  "cteed_landmark_driftwood_shade": "0",
33
  "cteed_landmark_driftwood_sun": "1",
 
47
  },
48
  "layer_norm_eps": 1e-12,
49
  "model_type": "vit",
50
+ "num_attention_heads": 12,
51
  "num_channels": 3,
52
+ "num_hidden_layers": 12,
53
  "patch_size": 16,
54
  "qkv_bias": true,
55
  "transformers_version": "4.38.2"
preprocessor_config.json CHANGED
@@ -16,7 +16,7 @@
16
  "resample": 2,
17
  "rescale_factor": 0.00392156862745098,
18
  "size": {
19
- "height": 224,
20
- "width": 224
21
  }
22
  }
 
16
  "resample": 2,
17
  "rescale_factor": 0.00392156862745098,
18
  "size": {
19
+ "height": 384,
20
+ "width": 384
21
  }
22
  }
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:809e00f453d3c66b4856e10a2c954417b439883d3519667b95cfb38bd7e28c92
3
- size 1213816912
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0f6e05741eb47ffdbac2a905a17d7a64b3e946fe3e5caff0a470827ffcce0e42
3
+ size 344677048