Hugging Face's logo

as-cle-bert
/
bcus-class-segformer

Image Classification
Transformers
TensorBoard
Safetensors
segformer
Trained with AutoTrain
Inference Endpoints
Model card Files Metrics Community
as-cle-bert commited on
Commit
071fa35
•
1 Parent(s): dce94f2

Upload folder using huggingface_hub

Browse files
Files changed (7) hide show
  1. README.md +22 -0
  2. config.json +80 -0
  3. model.safetensors +3 -0
  4. preprocessor_config.json +39 -0
  5. runs/Apr29_13-02-16_r-as-cle-bert-percbreac-is5m8hh9-3ecfd-n15uf/events.out.tfevents.1714395737.r-as-cle-bert-percbreac-is5m8hh9-3ecfd-n15uf.100.0 +2 -2
  6. training_args.bin +3 -0
  7. training_params.json +27 -0
README.md ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ tags:
4
+ - autotrain
5
+ - image-classification
6
+ widget:
7
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg
8
+ example_title: Tiger
9
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/teapot.jpg
10
+ example_title: Teapot
11
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/palace.jpg
12
+ example_title: Palace
13
+ datasets:
14
+ - as-cle-bert/breastcanc-ultrasound-class
15
+ ---
16
+
17
+ # Model Trained Using AutoTrain
18
+
19
+ - Problem type: Image Classification
20
+
21
+ ## Validation Metrics
22
+ No validation metrics available
config.json ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "nvidia/mit-b2",
3
+ "_num_labels": 2,
4
+ "architectures": [
5
+ "SegformerForImageClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.0,
8
+ "classifier_dropout_prob": 0.1,
9
+ "decoder_hidden_size": 768,
10
+ "depths": [
11
+ 3,
12
+ 4,
13
+ 6,
14
+ 3
15
+ ],
16
+ "downsampling_rates": [
17
+ 1,
18
+ 4,
19
+ 8,
20
+ 16
21
+ ],
22
+ "drop_path_rate": 0.1,
23
+ "hidden_act": "gelu",
24
+ "hidden_dropout_prob": 0.0,
25
+ "hidden_sizes": [
26
+ 64,
27
+ 128,
28
+ 320,
29
+ 512
30
+ ],
31
+ "id2label": {
32
+ "0": "benign_breast_cancer",
33
+ "1": "malignant_breast_cancer"
34
+ },
35
+ "image_size": 224,
36
+ "initializer_range": 0.02,
37
+ "label2id": {
38
+ "benign_breast_cancer": 0,
39
+ "malignant_breast_cancer": 1
40
+ },
41
+ "layer_norm_eps": 1e-06,
42
+ "mlp_ratios": [
43
+ 4,
44
+ 4,
45
+ 4,
46
+ 4
47
+ ],
48
+ "model_type": "segformer",
49
+ "num_attention_heads": [
50
+ 1,
51
+ 2,
52
+ 5,
53
+ 8
54
+ ],
55
+ "num_channels": 3,
56
+ "num_encoder_blocks": 4,
57
+ "patch_sizes": [
58
+ 7,
59
+ 3,
60
+ 3,
61
+ 3
62
+ ],
63
+ "problem_type": "single_label_classification",
64
+ "reshape_last_stage": true,
65
+ "semantic_loss_ignore_index": 255,
66
+ "sr_ratios": [
67
+ 8,
68
+ 4,
69
+ 2,
70
+ 1
71
+ ],
72
+ "strides": [
73
+ 4,
74
+ 2,
75
+ 2,
76
+ 2
77
+ ],
78
+ "torch_dtype": "float32",
79
+ "transformers_version": "4.40.1"
80
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:616b89dd376bb3b7405ff372a277d9f1280da9d02728aa62e2edb847813fcebf
3
+ size 96832704
preprocessor_config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_valid_processor_keys": [
3
+ "images",
4
+ "segmentation_maps",
5
+ "do_resize",
6
+ "size",
7
+ "resample",
8
+ "do_rescale",
9
+ "rescale_factor",
10
+ "do_normalize",
11
+ "image_mean",
12
+ "image_std",
13
+ "do_reduce_labels",
14
+ "return_tensors",
15
+ "data_format",
16
+ "input_data_format"
17
+ ],
18
+ "do_normalize": true,
19
+ "do_reduce_labels": false,
20
+ "do_rescale": true,
21
+ "do_resize": true,
22
+ "image_mean": [
23
+ 0.485,
24
+ 0.456,
25
+ 0.406
26
+ ],
27
+ "image_processor_type": "SegformerImageProcessor",
28
+ "image_std": [
29
+ 0.229,
30
+ 0.224,
31
+ 0.225
32
+ ],
33
+ "resample": 2,
34
+ "rescale_factor": 0.00392156862745098,
35
+ "size": {
36
+ "height": 512,
37
+ "width": 512
38
+ }
39
+ }
runs/Apr29_13-02-16_r-as-cle-bert-percbreac-is5m8hh9-3ecfd-n15uf/events.out.tfevents.1714395737.r-as-cle-bert-percbreac-is5m8hh9-3ecfd-n15uf.100.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6c6e12cdc96c8a8d069fb6bf1703c6590cac49391714fe8393a1e81a6e836be8
3
- size 8175
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77e4e7fe2d1c7d375573a071e0630c87d447a38e925e1628ff5ea8893155a569
3
+ size 8740
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8c20771870e564a7d4eb4b29bce6de9e0da4f5fa3c7e05858a1873e5f94162c
3
+ size 5048
training_params.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "as-cle-bert/breastcanc-ultrasound-class",
3
+ "model": "nvidia/mit-b2",
4
+ "username": "as-cle-bert",
5
+ "lr": 5e-05,
6
+ "epochs": 3,
7
+ "batch_size": 8,
8
+ "warmup_ratio": 0.1,
9
+ "gradient_accumulation": 1,
10
+ "optimizer": "adamw_torch",
11
+ "scheduler": "linear",
12
+ "weight_decay": 0.0,
13
+ "max_grad_norm": 1.0,
14
+ "seed": 42,
15
+ "train_split": "train",
16
+ "valid_split": null,
17
+ "logging_steps": -1,
18
+ "project_name": "autotrain-wfgvn-w4ij9",
19
+ "auto_find_batch_size": false,
20
+ "mixed_precision": "no",
21
+ "save_total_limit": 1,
22
+ "push_to_hub": true,
23
+ "evaluation_strategy": "epoch",
24
+ "image_column": "image",
25
+ "target_column": "label",
26
+ "log": "tensorboard"
27
+ }