Hugging Face's logo

Emilio407
/
prostate-mri-T2w-v03

Image Classification
TensorBoard
Safetensors
resnet
Trained with AutoTrain
Model card Files Metrics Community
Emilio407 commited on
Commit
a72fe43
•
1 Parent(s): 2516905

Upload folder using huggingface_hub

Browse files
Files changed (15) hide show
  1. README.md +31 -0
  2. checkpoint-528/config.json +50 -0
  3. checkpoint-528/model.safetensors +3 -0
  4. checkpoint-528/optimizer.pt +3 -0
  5. checkpoint-528/rng_state.pth +3 -0
  6. checkpoint-528/scheduler.pt +3 -0
  7. checkpoint-528/trainer_state.json +0 -0
  8. checkpoint-528/training_args.bin +3 -0
  9. config.json +50 -0
  10. model.safetensors +3 -0
  11. preprocessor_config.json +22 -0
  12. runs/Aug17_22-41-21_tufta/events.out.tfevents.1723948881.tufta.66785.0 +2 -2
  13. runs/Aug17_22-41-21_tufta/events.out.tfevents.1723949877.tufta.66785.1 +3 -0
  14. training_args.bin +3 -0
  15. training_params.json +29 -0
README.md ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ tags:
4
+ - autotrain
5
+ - image-classification
6
+ base_model: microsoft/resnet-152
7
+ widget:
8
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg
9
+ example_title: Tiger
10
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/teapot.jpg
11
+ example_title: Teapot
12
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/palace.jpg
13
+ example_title: Palace
14
+ ---
15
+
16
+ # Model Trained Using AutoTrain
17
+
18
+ - Problem type: Image Classification
19
+
20
+ ## Validation Metrics
21
+ loss: 0.2574561536312103
22
+
23
+ f1: 0.8857142857142857
24
+
25
+ precision: 0.8611111111111112
26
+
27
+ recall: 0.9117647058823529
28
+
29
+ auc: 0.9705882352941176
30
+
31
+ accuracy: 0.8709677419354839
checkpoint-528/config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "microsoft/resnet-152",
3
+ "_num_labels": 2,
4
+ "architectures": [
5
+ "ResNetForImageClassification"
6
+ ],
7
+ "depths": [
8
+ 3,
9
+ 8,
10
+ 36,
11
+ 3
12
+ ],
13
+ "downsample_in_bottleneck": false,
14
+ "downsample_in_first_stage": false,
15
+ "embedding_size": 64,
16
+ "hidden_act": "relu",
17
+ "hidden_sizes": [
18
+ 256,
19
+ 512,
20
+ 1024,
21
+ 2048
22
+ ],
23
+ "id2label": {
24
+ "0": "no",
25
+ "1": "yes"
26
+ },
27
+ "label2id": {
28
+ "no": 0,
29
+ "yes": 1
30
+ },
31
+ "layer_type": "bottleneck",
32
+ "model_type": "resnet",
33
+ "num_channels": 3,
34
+ "out_features": [
35
+ "stage4"
36
+ ],
37
+ "out_indices": [
38
+ 4
39
+ ],
40
+ "problem_type": "single_label_classification",
41
+ "stage_names": [
42
+ "stem",
43
+ "stage1",
44
+ "stage2",
45
+ "stage3",
46
+ "stage4"
47
+ ],
48
+ "torch_dtype": "float32",
49
+ "transformers_version": "4.44.0"
50
+ }
checkpoint-528/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc7534823c74ec03f815a812113946fb118e635eaeb3e9f150ec86fffc964f6a
3
+ size 233321808
checkpoint-528/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e80a08c05c3fd09f704138096c3f95d9e448ab403d1c11d77bebfb807d69e0a
3
+ size 465576843
checkpoint-528/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a56140d00d6505d9886ddff2cc36fd9ce60e0cdd7efe007991be987080faf32f
3
+ size 14180
checkpoint-528/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb237d130829bdad9e6ff9c6ff7a799936e4841bd3fc1e232af4d05aa0fccffe
3
+ size 1064
checkpoint-528/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-528/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:43c58db48412758b06c560cc345398b220af8c8b3e2a28e2353c6be42c7f1bad
3
+ size 5176
config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "microsoft/resnet-152",
3
+ "_num_labels": 2,
4
+ "architectures": [
5
+ "ResNetForImageClassification"
6
+ ],
7
+ "depths": [
8
+ 3,
9
+ 8,
10
+ 36,
11
+ 3
12
+ ],
13
+ "downsample_in_bottleneck": false,
14
+ "downsample_in_first_stage": false,
15
+ "embedding_size": 64,
16
+ "hidden_act": "relu",
17
+ "hidden_sizes": [
18
+ 256,
19
+ 512,
20
+ 1024,
21
+ 2048
22
+ ],
23
+ "id2label": {
24
+ "0": "no",
25
+ "1": "yes"
26
+ },
27
+ "label2id": {
28
+ "no": 0,
29
+ "yes": 1
30
+ },
31
+ "layer_type": "bottleneck",
32
+ "model_type": "resnet",
33
+ "num_channels": 3,
34
+ "out_features": [
35
+ "stage4"
36
+ ],
37
+ "out_indices": [
38
+ 4
39
+ ],
40
+ "problem_type": "single_label_classification",
41
+ "stage_names": [
42
+ "stem",
43
+ "stage1",
44
+ "stage2",
45
+ "stage3",
46
+ "stage4"
47
+ ],
48
+ "torch_dtype": "float32",
49
+ "transformers_version": "4.44.0"
50
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc7534823c74ec03f815a812113946fb118e635eaeb3e9f150ec86fffc964f6a
3
+ size 233321808
preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "crop_pct": 0.875,
3
+ "do_normalize": true,
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": "ConvNextImageProcessor",
12
+ "image_std": [
13
+ 0.229,
14
+ 0.224,
15
+ 0.225
16
+ ],
17
+ "resample": 3,
18
+ "rescale_factor": 0.00392156862745098,
19
+ "size": {
20
+ "shortest_edge": 224
21
+ }
22
+ }
runs/Aug17_22-41-21_tufta/events.out.tfevents.1723948881.tufta.66785.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:edeb6d8b989004a543eda3bf68b82d0159256e0c5e87302955665d60394d77be
3
- size 110660
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:223055c007adf964ce35b7b088bf5f7ec5f05d439e3a0d7d915fc4841db2037b
3
+ size 172279
runs/Aug17_22-41-21_tufta/events.out.tfevents.1723949877.tufta.66785.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45458940f1e41aea99b33700be033d94a028765ff6de28e824dab2042277539d
3
+ size 607
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:43c58db48412758b06c560cc345398b220af8c8b3e2a28e2353c6be42c7f1bad
3
+ size 5176
training_params.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "prostate-mri-T2w-v03/autotrain-data",
3
+ "model": "microsoft/resnet-152",
4
+ "username": "Emilio407",
5
+ "lr": 5e-05,
6
+ "epochs": 50,
7
+ "batch_size": 16,
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": "validation",
17
+ "logging_steps": -1,
18
+ "project_name": "prostate-mri-T2w-v03",
19
+ "auto_find_batch_size": false,
20
+ "mixed_precision": "bf16",
21
+ "save_total_limit": 1,
22
+ "push_to_hub": true,
23
+ "eval_strategy": "epoch",
24
+ "image_column": "autotrain_image",
25
+ "target_column": "autotrain_label",
26
+ "log": "tensorboard",
27
+ "early_stopping_patience": 15,
28
+ "early_stopping_threshold": 0.01
29
+ }