Hugging Face's logo

tcvrishank
/
histo_train_segformer

Image Classification
Transformers
PyTorch
TensorBoard
segformer
Generated from Trainer
Eval Results
Inference Endpoints
Model card Files Metrics Community
1
tcvrishank commited on
Commit
e50c715
•
1 Parent(s): b7e7917

Training in progress, step 100

Browse files
Files changed (8) hide show
  1. config.json +59 -12
  2. preprocessor_config.json +10 -9
  3. pytorch_model.bin +2 -2
  4. runs/Mar27_21-45-18_1d680bb46753/1679953650.0423949/events.out.tfevents.1679953650.1d680bb46753.997.1 +3 -0
  5. runs/Mar27_21-45-18_1d680bb46753/events.out.tfevents.1679953650.1d680bb46753.997.0 +3 -0
  6. runs/Mar27_21-48-33_1d680bb46753/1679953724.1269505/events.out.tfevents.1679953724.1d680bb46753.997.3 +3 -0
  7. runs/Mar27_21-48-33_1d680bb46753/events.out.tfevents.1679953724.1d680bb46753.997.2 +3 -0
  8. training_args.bin +2 -2
config.json CHANGED
@@ -1,13 +1,32 @@
1
  {
2
- "_name_or_path": "tcvrishank/histo_train_vit",
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": "Benign",
13
  "1": "InSitu",
@@ -16,21 +35,49 @@
16
  },
17
  "image_size": 224,
18
  "initializer_range": 0.02,
19
- "intermediate_size": 3072,
20
  "label2id": {
21
  "Benign": "0",
22
  "InSitu": "1",
23
  "Invasive": "2",
24
  "Normal": "3"
25
  },
26
- "layer_norm_eps": 1e-12,
27
- "model_type": "vit",
28
- "num_attention_heads": 12,
 
 
 
 
 
 
 
 
 
 
 
29
  "num_channels": 3,
30
- "num_hidden_layers": 12,
31
- "patch_size": 16,
 
 
 
 
 
32
  "problem_type": "single_label_classification",
33
- "qkv_bias": true,
 
 
 
 
 
 
 
 
 
 
 
 
 
34
  "torch_dtype": "float32",
35
- "transformers_version": "4.26.1"
36
  }
 
1
  {
2
+ "_name_or_path": "nvidia/mit-b0",
3
  "architectures": [
4
+ "SegformerForImageClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 256,
9
+ "depths": [
10
+ 2,
11
+ 2,
12
+ 2,
13
+ 2
14
+ ],
15
+ "downsampling_rates": [
16
+ 1,
17
+ 4,
18
+ 8,
19
+ 16
20
+ ],
21
+ "drop_path_rate": 0.1,
22
  "hidden_act": "gelu",
23
  "hidden_dropout_prob": 0.0,
24
+ "hidden_sizes": [
25
+ 32,
26
+ 64,
27
+ 160,
28
+ 256
29
+ ],
30
  "id2label": {
31
  "0": "Benign",
32
  "1": "InSitu",
 
35
  },
36
  "image_size": 224,
37
  "initializer_range": 0.02,
 
38
  "label2id": {
39
  "Benign": "0",
40
  "InSitu": "1",
41
  "Invasive": "2",
42
  "Normal": "3"
43
  },
44
+ "layer_norm_eps": 1e-06,
45
+ "mlp_ratios": [
46
+ 4,
47
+ 4,
48
+ 4,
49
+ 4
50
+ ],
51
+ "model_type": "segformer",
52
+ "num_attention_heads": [
53
+ 1,
54
+ 2,
55
+ 5,
56
+ 8
57
+ ],
58
  "num_channels": 3,
59
+ "num_encoder_blocks": 4,
60
+ "patch_sizes": [
61
+ 7,
62
+ 3,
63
+ 3,
64
+ 3
65
+ ],
66
  "problem_type": "single_label_classification",
67
+ "reshape_last_stage": true,
68
+ "semantic_loss_ignore_index": 255,
69
+ "sr_ratios": [
70
+ 8,
71
+ 4,
72
+ 2,
73
+ 1
74
+ ],
75
+ "strides": [
76
+ 4,
77
+ 2,
78
+ 2,
79
+ 2
80
+ ],
81
  "torch_dtype": "float32",
82
+ "transformers_version": "4.27.3"
83
  }
preprocessor_config.json CHANGED
@@ -1,22 +1,23 @@
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": "ViTFeatureExtractor",
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
  "do_normalize": true,
3
+ "do_reduce_labels": false,
4
  "do_rescale": true,
5
  "do_resize": true,
6
  "image_mean": [
7
+ 0.485,
8
+ 0.456,
9
+ 0.406
10
  ],
11
+ "image_processor_type": "SegformerImageProcessor",
12
  "image_std": [
13
+ 0.229,
14
+ 0.224,
15
+ 0.225
16
  ],
17
  "resample": 2,
18
  "rescale_factor": 0.00392156862745098,
19
  "size": {
20
+ "height": 512,
21
+ "width": 512
22
  }
23
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:52b81b9436f28f788ea36ab0eaab1e7c2e9cc6cd58230462018bb07f2cd552fc
3
- size 343274861
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a6e25292f23f17a9d9501b4d86f90f2eb3b2ddc939eb75bb5ca86fde7e2fffb
3
+ size 13348117
runs/Mar27_21-45-18_1d680bb46753/1679953650.0423949/events.out.tfevents.1679953650.1d680bb46753.997.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fa47928c54d01f95602e893b04908c82e7a849b46435d23fea53d829e6bd5caa
3
+ size 5800
runs/Mar27_21-45-18_1d680bb46753/events.out.tfevents.1679953650.1d680bb46753.997.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d9e45031524f582c02c4f6759ceb18c868021069c51edbc4140eb84ae0a8b7e4
3
+ size 4477
runs/Mar27_21-48-33_1d680bb46753/1679953724.1269505/events.out.tfevents.1679953724.1d680bb46753.997.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:efe26875a7fd59809cf0000d937ed31cc25067c76233f7d4250f3e99bd8779a0
3
+ size 5800
runs/Mar27_21-48-33_1d680bb46753/events.out.tfevents.1679953724.1d680bb46753.997.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50c6f2237dab965f06c7c2b08db3029e9afdd4def80e2d73cb7e765786c4744e
3
+ size 6334
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:51c784bcfe84093acfb329ab8cdd8aa46a6c4c20bd94e51587d2a6380e8a07ff
3
- size 3515
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5f9ccdb6e46ea1fdb7bdbaf84e86bb849fc81cd28016374a40073c7031972213
3
+ size 3579