initial commit

README.md

@@ -0,0 +1,77 @@
+---
+license: apache-2.0
+base_model: google/vit-base-patch16-224-in21k
+tags:
+- image-classification
+- generated_from_trainer
+datasets:
+- imagefolder
+metrics:
+- accuracy
+model-index:
+- name: vit-brain-tumour-v1
+ results:
+ - task:
+ name: Image Classification
+ type: image-classification
+ dataset:
+ name: essam24/brain-tumour-v1
+ type: imagefolder
+ config: default
+ split: test
+ args: default
+ metrics:
+ - name: Accuracy
+ type: accuracy
+ value: 1.0
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# vit-brain-tumour-v1
+
+This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the essam24/brain-tumour-v1 dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.0044
+- Accuracy: 1.0
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 0.0002
+- train_batch_size: 16
+- eval_batch_size: 8
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- num_epochs: 4
+- mixed_precision_training: Native AMP
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:------:|:----:|:---------------:|:--------:|
+| 0.0043 | 2.7027 | 100 | 0.0044 | 1.0 |
+
+
+### Framework versions
+
+- Transformers 4.42.4
+- Pytorch 2.4.0+cu121
+- Datasets 2.21.0
+- Tokenizers 0.19.1

all_results.json

@@ -0,0 +1,13 @@
+{
+ "epoch": 4.0,
+ "eval_accuracy": 1.0,
+ "eval_loss": 0.004365460481494665,
+ "eval_runtime": 3.862,
+ "eval_samples_per_second": 37.545,
+ "eval_steps_per_second": 4.92,
+ "total_flos": 1.800913838643118e+17,
+ "train_loss": 0.06353814376367105,
+ "train_runtime": 74.4207,
+ "train_samples_per_second": 31.228,
+ "train_steps_per_second": 1.989
+}

config.json

@@ -0,0 +1,32 @@
+{
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
+ "architectures": [
+ "ViTForImageClassification"
+ ],
+ "attention_probs_dropout_prob": 0.0,
+ "encoder_stride": 16,
+ "hidden_act": "gelu",
+ "hidden_dropout_prob": 0.0,
+ "hidden_size": 768,
+ "id2label": {
+ "0": "glioblastoma",
+ "1": "normal"
+ },
+ "image_size": 224,
+ "initializer_range": 0.02,
+ "intermediate_size": 3072,
+ "label2id": {
+ "glioblastoma": "0",
+ "normal": "1"
+ },
+ "layer_norm_eps": 1e-12,
+ "model_type": "vit",
+ "num_attention_heads": 12,
+ "num_channels": 3,
+ "num_hidden_layers": 12,
+ "patch_size": 16,
+ "problem_type": "single_label_classification",
+ "qkv_bias": true,
+ "torch_dtype": "float32",
+ "transformers_version": "4.42.4"
+}

eval_results.json

@@ -0,0 +1,8 @@
+{
+ "epoch": 4.0,
+ "eval_accuracy": 1.0,
+ "eval_loss": 0.004365460481494665,
+ "eval_runtime": 3.862,
+ "eval_samples_per_second": 37.545,
+ "eval_steps_per_second": 4.92
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ed2e20dd7e12eea67f170575fb1883d8483461104274623754ddcbe86c5f1b27
+size 343223968

preprocessor_config.json

@@ -0,0 +1,22 @@
+{
+ "do_normalize": true,
+ "do_rescale": true,
+ "do_resize": true,
+ "image_mean": [
+ 0.5,
+ 0.5,
+ 0.5
+ ],
+ "image_processor_type": "ViTImageProcessor",
+ "image_std": [
+ 0.5,
+ 0.5,
+ 0.5
+ ],
+ "resample": 2,
+ "rescale_factor": 0.00392156862745098,
+ "size": {
+ "height": 224,
+ "width": 224
+ }
+}

runs/Aug29_17-06-13_74a2be035470/events.out.tfevents.1724951185.74a2be035470.902.3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5706c3435f239f66df29ac221e001595078c1b4933495a9f0b63dfdc5bbb2575
+size 8429

runs/Aug29_17-06-13_74a2be035470/events.out.tfevents.1724951271.74a2be035470.902.4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eaea0ea0d8bb30dba756a2355071300551aa4ba81d7b62a21549eae4632f1989
+size 411

train_results.json

@@ -0,0 +1,8 @@
+{
+ "epoch": 4.0,
+ "total_flos": 1.800913838643118e+17,
+ "train_loss": 0.06353814376367105,
+ "train_runtime": 74.4207,
+ "train_samples_per_second": 31.228,
+ "train_steps_per_second": 1.989
+}

trainer_state.json

@@ -0,0 +1,149 @@
+{
+ "best_metric": 0.004365460481494665,
+ "best_model_checkpoint": "./vit-brain-tumour-v1/checkpoint-100",
+ "epoch": 4.0,
+ "eval_steps": 100,
+ "global_step": 148,
+ "is_hyper_param_search": false,
+ "is_local_process_zero": true,
+ "is_world_process_zero": true,
+ "log_history": [
+ {
+ "epoch": 0.2702702702702703,
+ "grad_norm": 2.4590463638305664,
+ "learning_rate": 0.0001864864864864865,
+ "loss": 0.4827,
+ "step": 10
+ },
+ {
+ "epoch": 0.5405405405405406,
+ "grad_norm": 2.9085402488708496,
+ "learning_rate": 0.000172972972972973,
+ "loss": 0.221,
+ "step": 20
+ },
+ {
+ "epoch": 0.8108108108108109,
+ "grad_norm": 0.13484928011894226,
+ "learning_rate": 0.00015945945945945947,
+ "loss": 0.0573,
+ "step": 30
+ },
+ {
+ "epoch": 1.0810810810810811,
+ "grad_norm": 3.143681287765503,
+ "learning_rate": 0.00014594594594594595,
+ "loss": 0.0655,
+ "step": 40
+ },
+ {
+ "epoch": 1.3513513513513513,
+ "grad_norm": 1.2873934507369995,
+ "learning_rate": 0.00013243243243243243,
+ "loss": 0.066,
+ "step": 50
+ },
+ {
+ "epoch": 1.6216216216216215,
+ "grad_norm": 0.05828818306326866,
+ "learning_rate": 0.00011891891891891893,
+ "loss": 0.009,
+ "step": 60
+ },
+ {
+ "epoch": 1.8918918918918919,
+ "grad_norm": 0.04554607346653938,
+ "learning_rate": 0.0001054054054054054,
+ "loss": 0.0063,
+ "step": 70
+ },
+ {
+ "epoch": 2.1621621621621623,
+ "grad_norm": 0.0411420576274395,
+ "learning_rate": 9.18918918918919e-05,
+ "loss": 0.0053,
+ "step": 80
+ },
+ {
+ "epoch": 2.4324324324324325,
+ "grad_norm": 0.03753970190882683,
+ "learning_rate": 7.837837837837838e-05,
+ "loss": 0.0048,
+ "step": 90
+ },
+ {
+ "epoch": 2.7027027027027026,
+ "grad_norm": 0.03439953923225403,
+ "learning_rate": 6.486486486486487e-05,
+ "loss": 0.0043,
+ "step": 100
+ },
+ {
+ "epoch": 2.7027027027027026,
+ "eval_accuracy": 1.0,
+ "eval_loss": 0.004365460481494665,
+ "eval_runtime": 2.8872,
+ "eval_samples_per_second": 50.222,
+ "eval_steps_per_second": 6.581,
+ "step": 100
+ },
+ {
+ "epoch": 2.972972972972973,
+ "grad_norm": 0.032108623534440994,
+ "learning_rate": 5.135135135135135e-05,
+ "loss": 0.0041,
+ "step": 110
+ },
+ {
+ "epoch": 3.2432432432432434,
+ "grad_norm": 0.03147277981042862,
+ "learning_rate": 3.783783783783784e-05,
+ "loss": 0.0038,
+ "step": 120
+ },
+ {
+ "epoch": 3.5135135135135136,
+ "grad_norm": 0.02981223724782467,
+ "learning_rate": 2.4324324324324327e-05,
+ "loss": 0.0037,
+ "step": 130
+ },
+ {
+ "epoch": 3.7837837837837838,
+ "grad_norm": 0.03125371038913727,
+ "learning_rate": 1.0810810810810812e-05,
+ "loss": 0.0037,
+ "step": 140
+ },
+ {
+ "epoch": 4.0,
+ "step": 148,
+ "total_flos": 1.800913838643118e+17,
+ "train_loss": 0.06353814376367105,
+ "train_runtime": 74.4207,
+ "train_samples_per_second": 31.228,
+ "train_steps_per_second": 1.989
+ }
+ ],
+ "logging_steps": 10,
+ "max_steps": 148,
+ "num_input_tokens_seen": 0,
+ "num_train_epochs": 4,
+ "save_steps": 100,
+ "stateful_callbacks": {
+ "TrainerControl": {
+ "args": {
+ "should_epoch_stop": false,
+ "should_evaluate": false,
+ "should_log": false,
+ "should_save": true,
+ "should_training_stop": true
+ },
+ "attributes": {}
+ }
+ },
+ "total_flos": 1.800913838643118e+17,
+ "train_batch_size": 16,
+ "trial_name": null,
+ "trial_params": null
+}

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ec3c6906e79516bb436143f5f87786d3f02e5ea3acfeb64f8655ac2e1d877701
+size 5112