Update README.md
Browse files
README.md
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
---
|
2 |
library_name: transformers
|
3 |
-
tags: []
|
4 |
---
|
5 |
|
6 |
# Model Card for Model ID
|
@@ -77,13 +77,14 @@ Use the code below to get started with the model.
|
|
77 |
|
78 |
### Training Data
|
79 |
|
80 |
-
|
81 |
|
82 |
[More Information Needed]
|
83 |
|
84 |
### Training Procedure
|
85 |
|
86 |
-
|
|
|
87 |
|
88 |
#### Preprocessing [optional]
|
89 |
|
|
|
1 |
---
|
2 |
library_name: transformers
|
3 |
+
tags: [Image Classification, MRI, Cancer]
|
4 |
---
|
5 |
|
6 |
# Model Card for Model ID
|
|
|
77 |
|
78 |
### Training Data
|
79 |
|
80 |
+
https://huggingface.co/datasets/tanzuhuggingface/brainmri
|
81 |
|
82 |
[More Information Needed]
|
83 |
|
84 |
### Training Procedure
|
85 |
|
86 |
+
The restnet18 model was fin-tuned with P100 GPU for 200 epochs. Both calibration and validation losses decined constantly during the fine-tuning showing no sign of overfitting.
|
87 |
+
The final accuracy was 97.9%.
|
88 |
|
89 |
#### Preprocessing [optional]
|
90 |
|