Hugging Face's logo

Binou
/
vit-base-plankton

Image Classification
Transformers
TensorBoard
Safetensors
mobilevit
Generated from Trainer
Eval Results
Inference Endpoints
Model card Files Metrics Community
Binou commited on
Commit
d4c9ed8
•
1 Parent(s): d3d8db2

Training in progress, step 100

Browse files
Files changed (6) hide show
  1. config.json +55 -33
  2. model.safetensors +2 -2
  3. preprocessor_config.json +12 -15
  4. runs/Nov19_15-34-24_12be7f682407/events.out.tfevents.1700408194.12be7f682407.128.0 +3 -0
  5. runs/Nov19_15-47-32_12be7f682407/events.out.tfevents.1700408857.12be7f682407.128.1 +3 -0
  6. training_args.bin +1 -1
config.json CHANGED
@@ -1,55 +1,77 @@
1
  {
2
- "_name_or_path": "google/vit-base-patch16-224-in21k",
3
  "architectures": [
4
- "ViTForImageClassification"
 
 
 
 
 
 
 
5
  ],
6
  "attention_probs_dropout_prob": 0.0,
7
- "encoder_stride": 16,
8
- "hidden_act": "gelu",
9
- "hidden_dropout_prob": 0.0,
10
- "hidden_size": 768,
 
 
 
 
 
 
11
  "id2label": {
12
  "0": "Acantharia",
13
  "1": "Calanoida",
14
- "10": "Rhizaria",
15
- "11": "Rhizosolenia",
16
- "12": "Thalassiosira",
17
- "13": "nauplii_Copepoda",
18
  "2": "Chaetoceros_Chaetocerotales",
19
  "3": "Ciliophora",
20
- "4": "Coscinodiscophycidae",
21
- "5": "Eutintinnus",
22
- "6": "Navicula",
23
- "7": "Neoceratium",
24
- "8": "Planktoniella_sol",
25
- "9": "Pyrophacus"
26
  },
27
- "image_size": 224,
28
  "initializer_range": 0.02,
29
- "intermediate_size": 3072,
30
  "label2id": {
31
  "Acantharia": "0",
32
  "Calanoida": "1",
33
  "Chaetoceros_Chaetocerotales": "2",
34
  "Ciliophora": "3",
35
- "Coscinodiscophycidae": "4",
36
- "Eutintinnus": "5",
37
- "Navicula": "6",
38
- "Neoceratium": "7",
39
- "Planktoniella_sol": "8",
40
- "Pyrophacus": "9",
41
- "Rhizaria": "10",
42
- "Rhizosolenia": "11",
43
- "Thalassiosira": "12",
44
- "nauplii_Copepoda": "13"
45
  },
46
- "layer_norm_eps": 1e-12,
47
- "model_type": "vit",
48
- "num_attention_heads": 12,
 
 
 
 
 
 
 
 
 
 
49
  "num_channels": 3,
50
- "num_hidden_layers": 12,
51
- "patch_size": 16,
 
52
  "qkv_bias": true,
 
53
  "torch_dtype": "float32",
54
  "transformers_version": "4.35.2"
55
  }
 
1
  {
2
+ "_name_or_path": "apple/mobilevit-xx-small",
3
  "architectures": [
4
+ "MobileViTForImageClassification"
5
+ ],
6
+ "aspp_dropout_prob": 0.1,
7
+ "aspp_out_channels": 256,
8
+ "atrous_rates": [
9
+ 6,
10
+ 12,
11
+ 18
12
  ],
13
  "attention_probs_dropout_prob": 0.0,
14
+ "classifier_dropout_prob": 0.1,
15
+ "conv_kernel_size": 3,
16
+ "expand_ratio": 2.0,
17
+ "hidden_act": "silu",
18
+ "hidden_dropout_prob": 0.05,
19
+ "hidden_sizes": [
20
+ 64,
21
+ 80,
22
+ 96
23
+ ],
24
  "id2label": {
25
  "0": "Acantharia",
26
  "1": "Calanoida",
27
+ "10": "Rhizosolenia",
28
+ "11": "Thalassiosira",
29
+ "12": "nauplii_Copepoda",
 
30
  "2": "Chaetoceros_Chaetocerotales",
31
  "3": "Ciliophora",
32
+ "4": "Eutintinnus",
33
+ "5": "Navicula",
34
+ "6": "Neoceratium",
35
+ "7": "Planktoniella_sol",
36
+ "8": "Pyrophacus",
37
+ "9": "Rhizaria"
38
  },
39
+ "image_size": 256,
40
  "initializer_range": 0.02,
 
41
  "label2id": {
42
  "Acantharia": "0",
43
  "Calanoida": "1",
44
  "Chaetoceros_Chaetocerotales": "2",
45
  "Ciliophora": "3",
46
+ "Eutintinnus": "4",
47
+ "Navicula": "5",
48
+ "Neoceratium": "6",
49
+ "Planktoniella_sol": "7",
50
+ "Pyrophacus": "8",
51
+ "Rhizaria": "9",
52
+ "Rhizosolenia": "10",
53
+ "Thalassiosira": "11",
54
+ "nauplii_Copepoda": "12"
 
55
  },
56
+ "layer_norm_eps": 1e-05,
57
+ "mlp_ratio": 2.0,
58
+ "model_type": "mobilevit",
59
+ "neck_hidden_sizes": [
60
+ 16,
61
+ 16,
62
+ 24,
63
+ 48,
64
+ 64,
65
+ 80,
66
+ 320
67
+ ],
68
+ "num_attention_heads": 4,
69
  "num_channels": 3,
70
+ "output_stride": 32,
71
+ "patch_size": 2,
72
+ "problem_type": "single_label_classification",
73
  "qkv_bias": true,
74
+ "semantic_loss_ignore_index": 255,
75
  "torch_dtype": "float32",
76
  "transformers_version": "4.35.2"
77
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:133b10f5e46dbc59da71c447a0bbe6cef3e08afe98eadc74876d5f4d180f2459
3
- size 343260888
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1fd62dd256bfe9875c42882967f294ec417366d28a7869f8aa686371f583682e
3
+ size 3883484
preprocessor_config.json CHANGED
@@ -1,22 +1,19 @@
1
  {
2
- "do_normalize": true,
 
 
 
 
 
 
3
  "do_rescale": true,
4
  "do_resize": true,
5
- "image_mean": [
6
- 0.5,
7
- 0.5,
8
- 0.5
9
- ],
10
- "image_processor_type": "ViTImageProcessor",
11
- "image_std": [
12
- 0.5,
13
- 0.5,
14
- 0.5
15
- ],
16
  "resample": 2,
17
  "rescale_factor": 0.00392156862745098,
18
  "size": {
19
- "height": 224,
20
- "width": 224
21
- }
22
  }
 
1
  {
2
+ "crop_size": {
3
+ "height": 256,
4
+ "width": 256
5
+ },
6
+ "do_center_crop": true,
7
+ "do_flip_channel_order": true,
8
+ "do_flip_channels": true,
9
  "do_rescale": true,
10
  "do_resize": true,
11
+ "feature_extractor_type": "MobileViTFeatureExtractor",
12
+ "image_processor_type": "MobileViTFeatureExtractor",
 
 
 
 
 
 
 
 
 
13
  "resample": 2,
14
  "rescale_factor": 0.00392156862745098,
15
  "size": {
16
+ "shortest_edge": 288
17
+ },
18
+ "use_square_size": false
19
  }
runs/Nov19_15-34-24_12be7f682407/events.out.tfevents.1700408194.12be7f682407.128.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cd9d7059bd867adcd5673bb738890d7cbc8123aec06197724496a8467865f0c0
3
+ size 5146
runs/Nov19_15-47-32_12be7f682407/events.out.tfevents.1700408857.12be7f682407.128.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8bbf2487d80d4fcec03bfbbe0c2a1050e60ad868db78b5474fdf8984830ecb06
3
+ size 7003
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fdce132aac6f6ff84a8527cab62067c32eb21a6fb26be8e3c1cf8e5976305f09
3
  size 4536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68485bc6b00c179249405697fb98f89cfdd94f2bc8f75e9e6a75dbd190524856
3
  size 4536