synergyai-jaeung commited on
Commit
f4371d1
1 Parent(s): 07349ee

Model save

Browse files
README.md ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: DunnBC22/vit-base-patch16-224-in21k_covid_19_ct_scans
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ - precision
9
+ - recall
10
+ - f1
11
+ model-index:
12
+ - name: vit-base-patch16-224-in21k_covid_19_ct_scans-finetuned-RCC
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # vit-base-patch16-224-in21k_covid_19_ct_scans-finetuned-RCC
20
+
21
+ This model is a fine-tuned version of [DunnBC22/vit-base-patch16-224-in21k_covid_19_ct_scans](https://huggingface.co/DunnBC22/vit-base-patch16-224-in21k_covid_19_ct_scans) on the None dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 0.3235
24
+ - Accuracy: 0.9032
25
+ - Precision: 0.9032
26
+ - Recall: 1.0
27
+ - F1: 0.4746
28
+ - Auc: 0.5
29
+
30
+ ## Model description
31
+
32
+ More information needed
33
+
34
+ ## Intended uses & limitations
35
+
36
+ More information needed
37
+
38
+ ## Training and evaluation data
39
+
40
+ More information needed
41
+
42
+ ## Training procedure
43
+
44
+ ### Training hyperparameters
45
+
46
+ The following hyperparameters were used during training:
47
+ - learning_rate: 5e-05
48
+ - train_batch_size: 32
49
+ - eval_batch_size: 32
50
+ - seed: 42
51
+ - gradient_accumulation_steps: 4
52
+ - total_train_batch_size: 128
53
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
54
+ - lr_scheduler_type: linear
55
+ - lr_scheduler_warmup_ratio: 0.1
56
+ - num_epochs: 5
57
+
58
+ ### Training results
59
+
60
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 | Auc |
61
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|:---:|
62
+ | No log | 1.0 | 7 | 0.3327 | 0.9032 | 0.9032 | 1.0 | 0.4746 | 0.5 |
63
+ | 0.3866 | 2.0 | 14 | 0.3213 | 0.9032 | 0.9032 | 1.0 | 0.4746 | 0.5 |
64
+ | 0.2647 | 3.0 | 21 | 0.3226 | 0.9032 | 0.9032 | 1.0 | 0.4746 | 0.5 |
65
+ | 0.2647 | 4.0 | 28 | 0.3246 | 0.9032 | 0.9032 | 1.0 | 0.4746 | 0.5 |
66
+ | 0.2593 | 5.0 | 35 | 0.3235 | 0.9032 | 0.9032 | 1.0 | 0.4746 | 0.5 |
67
+
68
+
69
+ ### Framework versions
70
+
71
+ - Transformers 4.41.1
72
+ - Pytorch 2.0.0+cu117
73
+ - Datasets 2.19.1
74
+ - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:29f228df3d99ae7ebf5e6defca6ef82583f16d3b233c75232c8398160110f2e0
3
  size 343223968
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf579d8d432897475479bd8b99f9b0e80634bf5f6015cbaf759efc00d151cd74
3
  size 343223968
runs/May27_17-11-52_RTX3090/events.out.tfevents.1716797516.RTX3090.8398.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a5be2460dc82e26b5db90968c0888c95388ec059e2817a7b038801df3e4e6c87
3
- size 7506
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fbe216952598c83d9f9af21613ed6a089f52b5112bd10d53b9e76cccd6075997
3
+ size 8570