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