Hugging Face's logo

diegokauer
/
int-pet-classifier

Image Classification
Transformers
TensorBoard
Safetensors
vit
Inference Endpoints
Model card Files Metrics Community
diegokauer commited on
Commit
72578d0
•
1 Parent(s): af6326c

Training in progress, epoch 0

Browse files
Files changed (27) hide show
  1. config.json +56 -73
  2. model.safetensors +2 -2
  3. preprocessor_config.json +10 -9
  4. runs/Jan02_15-46-16_00d62c54e1af/events.out.tfevents.1704210377.00d62c54e1af.378.10 +2 -2
  5. runs/Jan02_17-26-32_00d62c54e1af/events.out.tfevents.1704216393.00d62c54e1af.378.11 +3 -0
  6. runs/Jan02_17-45-55_00d62c54e1af/events.out.tfevents.1704217556.00d62c54e1af.378.12 +3 -0
  7. runs/Jan02_17-55-14_00d62c54e1af/events.out.tfevents.1704218115.00d62c54e1af.378.13 +3 -0
  8. runs/Jan02_18-01-08_00d62c54e1af/events.out.tfevents.1704218469.00d62c54e1af.378.14 +3 -0
  9. runs/Jan02_18-01-33_00d62c54e1af/events.out.tfevents.1704218494.00d62c54e1af.378.15 +3 -0
  10. runs/Jan02_18-04-16_00d62c54e1af/events.out.tfevents.1704218657.00d62c54e1af.378.16 +3 -0
  11. runs/Jan02_18-08-44_00d62c54e1af/events.out.tfevents.1704218925.00d62c54e1af.378.17 +3 -0
  12. runs/Jan02_18-17-25_00d62c54e1af/events.out.tfevents.1704219446.00d62c54e1af.378.18 +3 -0
  13. runs/Jan02_18-20-03_00d62c54e1af/events.out.tfevents.1704219604.00d62c54e1af.378.19 +3 -0
  14. runs/Jan02_18-24-57_00d62c54e1af/events.out.tfevents.1704219898.00d62c54e1af.378.20 +3 -0
  15. runs/Jan02_18-29-32_00d62c54e1af/events.out.tfevents.1704220173.00d62c54e1af.378.21 +3 -0
  16. runs/Jan02_18-33-11_00d62c54e1af/events.out.tfevents.1704220392.00d62c54e1af.63979.0 +3 -0
  17. runs/Jan02_18-33-50_00d62c54e1af/events.out.tfevents.1704220432.00d62c54e1af.63979.1 +3 -0
  18. runs/Jan02_18-34-34_00d62c54e1af/events.out.tfevents.1704220477.00d62c54e1af.64502.0 +3 -0
  19. runs/Jan02_18-36-00_00d62c54e1af/events.out.tfevents.1704220561.00d62c54e1af.64502.1 +3 -0
  20. runs/Jan02_18-48-21_00d62c54e1af/events.out.tfevents.1704221303.00d62c54e1af.67958.0 +3 -0
  21. runs/Jan02_18-58-34_00d62c54e1af/events.out.tfevents.1704221915.00d62c54e1af.67958.1 +3 -0
  22. runs/Jan02_18-58-54_00d62c54e1af/events.out.tfevents.1704221934.00d62c54e1af.67958.2 +3 -0
  23. runs/Jan02_19-08-08_00d62c54e1af/events.out.tfevents.1704222489.00d62c54e1af.71418.0 +3 -0
  24. runs/Jan02_19-17-09_00d62c54e1af/events.out.tfevents.1704223030.00d62c54e1af.75304.0 +3 -0
  25. runs/Jan02_19-38-31_00d62c54e1af/events.out.tfevents.1704224312.00d62c54e1af.75304.1 +3 -0
  26. runs/Jan02_19-39-45_00d62c54e1af/events.out.tfevents.1704224387.00d62c54e1af.81026.0 +3 -0
  27. training_args.bin +1 -1
config.json CHANGED
@@ -1,88 +1,71 @@
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": "['No Reportado']",
13
- "1": "['Estudio Adicional']",
14
- "2": "['QMT']",
15
- "3": "['QMT', 'RadioTerapia']",
16
- "4": "['Otro']",
17
- "5": "['RadioTerapia']",
18
- "6": "['Cuidados Paliativos']",
19
- "7": "['Cirug\u00eda']",
20
- "8": "['QMT', 'RadioTerapia', 'Estudio Adicional']",
21
- "9": "['QMT', 'Estudio Adicional']",
22
- "10": "['RadioTerapia', 'Estudio Adicional']",
23
- "11": "['Cuidados Paliativos', 'Estudio Adicional']",
24
- "12": "['QMT', 'Otro']",
25
- "13": "['QMT', 'Cuidados Paliativos']",
26
- "14": "['Cirug\u00eda', 'Estudio Adicional']",
27
- "15": "['RadioTerapia', 'Otro']",
28
- "16": "['RadioTerapia', 'Cuidados Paliativos']",
29
- "17": "['QMT', 'RadioTerapia', 'Cuidados Paliativos']",
30
- "18": "['Estudio Adicional', 'Otro']",
31
- "19": "['Cuidados Paliativos', 'Otro']",
32
- "20": "['Cirug\u00eda', 'QMT']",
33
- "21": "['Cirug\u00eda', 'Otro']",
34
- "22": "['QMT', 'RadioTerapia', 'Otro']",
35
- "23": "['Cirug\u00eda', 'RadioTerapia']",
36
- "24": "['Cirug\u00eda', 'Estudio Adicional', 'Otro']",
37
- "25": "['QMT', 'Cuidados Paliativos', 'Otro']",
38
- "26": "['QMT', 'Estudio Adicional', 'Otro']",
39
- "27": "['RadioTerapia', 'Cuidados Paliativos', 'Otro']",
40
- "28": "['QMT', 'Cuidados Paliativos', 'Estudio Adicional']",
41
- "29": "['RadioTerapia', 'Cuidados Paliativos', 'Estudio Adicional']"
42
  },
43
- "image_size": 224,
44
  "initializer_range": 0.02,
45
- "intermediate_size": 3072,
46
  "label2id": {
47
- "['Cirug\u00eda', 'Estudio Adicional', 'Otro']": 24,
48
- "['Cirug\u00eda', 'Estudio Adicional']": 14,
49
- "['Cirug\u00eda', 'Otro']": 21,
50
- "['Cirug\u00eda', 'QMT']": 20,
51
- "['Cirug\u00eda', 'RadioTerapia']": 23,
52
- "['Cirug\u00eda']": 7,
53
- "['Cuidados Paliativos', 'Estudio Adicional']": 11,
54
- "['Cuidados Paliativos', 'Otro']": 19,
55
- "['Cuidados Paliativos']": 6,
56
- "['Estudio Adicional', 'Otro']": 18,
57
- "['Estudio Adicional']": 1,
58
- "['No Reportado']": 0,
59
- "['Otro']": 4,
60
- "['QMT', 'Cuidados Paliativos', 'Estudio Adicional']": 28,
61
- "['QMT', 'Cuidados Paliativos', 'Otro']": 25,
62
- "['QMT', 'Cuidados Paliativos']": 13,
63
- "['QMT', 'Estudio Adicional', 'Otro']": 26,
64
- "['QMT', 'Estudio Adicional']": 9,
65
- "['QMT', 'Otro']": 12,
66
- "['QMT', 'RadioTerapia', 'Cuidados Paliativos']": 17,
67
- "['QMT', 'RadioTerapia', 'Estudio Adicional']": 8,
68
- "['QMT', 'RadioTerapia', 'Otro']": 22,
69
- "['QMT', 'RadioTerapia']": 3,
70
- "['QMT']": 2,
71
- "['RadioTerapia', 'Cuidados Paliativos', 'Estudio Adicional']": 29,
72
- "['RadioTerapia', 'Cuidados Paliativos', 'Otro']": 27,
73
- "['RadioTerapia', 'Cuidados Paliativos']": 16,
74
- "['RadioTerapia', 'Estudio Adicional']": 10,
75
- "['RadioTerapia', 'Otro']": 15,
76
- "['RadioTerapia']": 5
77
  },
78
- "layer_norm_eps": 1e-12,
79
- "model_type": "vit",
80
- "num_attention_heads": 12,
81
  "num_channels": 3,
82
- "num_hidden_layers": 12,
83
- "patch_size": 16,
84
- "problem_type": "single_label_classification",
 
 
 
 
 
 
 
 
 
 
 
 
 
85
  "qkv_bias": true,
 
 
 
 
 
 
 
86
  "torch_dtype": "float32",
87
- "transformers_version": "4.35.2"
 
 
88
  }
 
1
  {
2
+ "_name_or_path": "microsoft/swin-base-patch4-window12-384-in22k",
3
  "architectures": [
4
+ "SwinForImageClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.0,
7
+ "depths": [
8
+ 2,
9
+ 2,
10
+ 18,
11
+ 2
12
+ ],
13
+ "drop_path_rate": 0.1,
14
+ "embed_dim": 128,
15
+ "encoder_stride": 32,
16
  "hidden_act": "gelu",
17
  "hidden_dropout_prob": 0.0,
18
+ "hidden_size": 1024,
19
  "id2label": {
20
+ "0": "No Reportado",
21
+ "1": "Cirug\u00eda",
22
+ "2": "QMT",
23
+ "3": "RadioTerapia",
24
+ "4": "Cuidados Paliativos",
25
+ "5": "Estudio Adicional",
26
+ "6": "Otro"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
27
  },
28
+ "image_size": 384,
29
  "initializer_range": 0.02,
 
30
  "label2id": {
31
+ "Cirug\u00eda": 1,
32
+ "Cuidados Paliativos": 4,
33
+ "Estudio Adicional": 5,
34
+ "No Reportado": 0,
35
+ "Otro": 6,
36
+ "QMT": 2,
37
+ "RadioTerapia": 3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
38
  },
39
+ "layer_norm_eps": 1e-05,
40
+ "mlp_ratio": 4.0,
41
+ "model_type": "swin",
42
  "num_channels": 3,
43
+ "num_heads": [
44
+ 4,
45
+ 8,
46
+ 16,
47
+ 32
48
+ ],
49
+ "num_layers": 4,
50
+ "out_features": [
51
+ "stage4"
52
+ ],
53
+ "out_indices": [
54
+ 4
55
+ ],
56
+ "patch_size": 4,
57
+ "path_norm": true,
58
+ "problem_type": "multi_label_classification",
59
  "qkv_bias": true,
60
+ "stage_names": [
61
+ "stem",
62
+ "stage1",
63
+ "stage2",
64
+ "stage3",
65
+ "stage4"
66
+ ],
67
  "torch_dtype": "float32",
68
+ "transformers_version": "4.35.2",
69
+ "use_absolute_embeddings": false,
70
+ "window_size": 12
71
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b7b8c0bb673622cc970a7263b1f04776c335c63b0377545f3ab8c6e612c4af0b
3
- size 343310104
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31de3a7e7ad6732e696857b1a449ac7ccdb55d2df434b22f156ff9131a9793e0
3
+ size 351581228
preprocessor_config.json CHANGED
@@ -2,21 +2,22 @@
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
  }
 
2
  "do_normalize": true,
3
  "do_rescale": true,
4
  "do_resize": true,
5
+ "feature_extractor_type": "ViTFeatureExtractor",
6
  "image_mean": [
7
+ 0.485,
8
+ 0.456,
9
+ 0.406
10
  ],
11
  "image_processor_type": "ViTImageProcessor",
12
  "image_std": [
13
+ 0.229,
14
+ 0.224,
15
+ 0.225
16
  ],
17
+ "resample": 3,
18
  "rescale_factor": 0.00392156862745098,
19
  "size": {
20
+ "height": 384,
21
+ "width": 384
22
  }
23
  }
runs/Jan02_15-46-16_00d62c54e1af/events.out.tfevents.1704210377.00d62c54e1af.378.10 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d3dfd355b4431edff52c3bb556f6e367ea3acc3ba3485c894300578518f8f928
3
- size 23855
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47a68504870049005bab8106b3e80e93e345edd90b860d981724401c280f5edd
3
+ size 25752
runs/Jan02_17-26-32_00d62c54e1af/events.out.tfevents.1704216393.00d62c54e1af.378.11 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8011f2d532b921b4e9e8fb778bc6f77cd163be5e9ceb9088e8c5759c958167b1
3
+ size 8374
runs/Jan02_17-45-55_00d62c54e1af/events.out.tfevents.1704217556.00d62c54e1af.378.12 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:848fc20a4aa302e1d3682cf19a61a8774c51832027fb5bd6d235bd6647763de6
3
+ size 1670328
runs/Jan02_17-55-14_00d62c54e1af/events.out.tfevents.1704218115.00d62c54e1af.378.13 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d53562f12d33d7bb46ce5df1349b01ddd925b0590d4507b97cb402bc09ddcd9
3
+ size 1670328
runs/Jan02_18-01-08_00d62c54e1af/events.out.tfevents.1704218469.00d62c54e1af.378.14 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:24f20ec5e001419ab5c9c088bf9bd15f60f70998fbe4e4c9caccb81d02be2a38
3
+ size 1670328
runs/Jan02_18-01-33_00d62c54e1af/events.out.tfevents.1704218494.00d62c54e1af.378.15 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a3bfbd8370022a874e05b4ad9c957798e4f1910c14273c7282df5c570573e10
3
+ size 1670328
runs/Jan02_18-04-16_00d62c54e1af/events.out.tfevents.1704218657.00d62c54e1af.378.16 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e6245cbc80d4cc48c48240f2b6b7504285a074a35ff53c06fa7fa59571705446
3
+ size 1670328
runs/Jan02_18-08-44_00d62c54e1af/events.out.tfevents.1704218925.00d62c54e1af.378.17 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3eae2419df2b0080ef6ba3dcca667a0d015e08e634a43e4f3d02938a116c088c
3
+ size 1670328
runs/Jan02_18-17-25_00d62c54e1af/events.out.tfevents.1704219446.00d62c54e1af.378.18 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:acf351f6e3b00962a245c71015ed8c5bfcf8968e9ab658332043b67e7f0f3d25
3
+ size 1670376
runs/Jan02_18-20-03_00d62c54e1af/events.out.tfevents.1704219604.00d62c54e1af.378.19 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c865a12f78b8ec51a60b037e412d2907237d8a483847dde747500361acbc84b3
3
+ size 1670376
runs/Jan02_18-24-57_00d62c54e1af/events.out.tfevents.1704219898.00d62c54e1af.378.20 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a30e66d27aa2122c8ac0218440a83b5e5741cd7a650b754751a14d6be761d84
3
+ size 1670376
runs/Jan02_18-29-32_00d62c54e1af/events.out.tfevents.1704220173.00d62c54e1af.378.21 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:69607899f7a01a9f3b213cbfd5686d83b245327b0345a6a24a3241cab3e18e3b
3
+ size 1670328
runs/Jan02_18-33-11_00d62c54e1af/events.out.tfevents.1704220392.00d62c54e1af.63979.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c21c84e34db5cf4e5936d68cb0a419d89acb4626df763421e6c753dd34cc27e
3
+ size 4184
runs/Jan02_18-33-50_00d62c54e1af/events.out.tfevents.1704220432.00d62c54e1af.63979.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cab5d33689f8b2dc5b62dbaaca48c53d96d40566cfe38462a971cea4f242d8f0
3
+ size 4184
runs/Jan02_18-34-34_00d62c54e1af/events.out.tfevents.1704220477.00d62c54e1af.64502.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ca79ccf37e33e59c6dd7af670a10d4ffb79489da639191186da28f5763d9d3f
3
+ size 4905
runs/Jan02_18-36-00_00d62c54e1af/events.out.tfevents.1704220561.00d62c54e1af.64502.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81469924799a45726f9fccd4703ae05199a926c16ebdd8dd51807d4c069b0cc7
3
+ size 6711
runs/Jan02_18-48-21_00d62c54e1af/events.out.tfevents.1704221303.00d62c54e1af.67958.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf60db63666c770aebde3ecb00839b64846d3aec2b66b66ff4b4f5f5abb401d0
3
+ size 5055
runs/Jan02_18-58-34_00d62c54e1af/events.out.tfevents.1704221915.00d62c54e1af.67958.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c41b8636914238bc9d82b7092349347da51b402200ecccaaddc43785442a7af7
3
+ size 4853
runs/Jan02_18-58-54_00d62c54e1af/events.out.tfevents.1704221934.00d62c54e1af.67958.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e9c57bd3f079f17159ac753e088a8af146bdcdbd423834ba3a60cad6c2838bc
3
+ size 5005
runs/Jan02_19-08-08_00d62c54e1af/events.out.tfevents.1704222489.00d62c54e1af.71418.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b061ea20bd5be9fee090d960600fcaa5167f3383f24b5f48aad472394c0cd32
3
+ size 5005
runs/Jan02_19-17-09_00d62c54e1af/events.out.tfevents.1704223030.00d62c54e1af.75304.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9954d7cf82e8fb3f43fdd6dc05584653d650a4c65f98c27b9e4a1b00f432d350
3
+ size 9253
runs/Jan02_19-38-31_00d62c54e1af/events.out.tfevents.1704224312.00d62c54e1af.75304.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:229bf917d226c648d2df074908ddc1cbe0c7b53174e991651d9cb64582595a35
3
+ size 4184
runs/Jan02_19-39-45_00d62c54e1af/events.out.tfevents.1704224387.00d62c54e1af.81026.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1069fb1dd9d485bb330f35e11c76b856aad1ebe5ca54b7ab815df5d719898dfe
3
+ size 6353
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0f1aa16383096cc3d1d7285aa4e604c8ff05d48166eeb964b07d541aaef1c90c
3
  size 4600
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ef45468732528c084d40fb77a6614f52abae50baf7cf124b00251f6c411c70e
3
  size 4600