Upload folder using huggingface_hub

checkpoint-12105/config.json

@@ -0,0 +1,34 @@
+{
+  "_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": "E",
+    "1": "S",
+    "2": "SB"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "E": 0,
+    "S": 1,
+    "SB": 2
+  },
+  "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.44.2"
+}

checkpoint-12105/model.safetensors

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

checkpoint-12105/optimizer.pt

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

checkpoint-12105/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
+  }
+}

checkpoint-12105/rng_state.pth

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

checkpoint-12105/scheduler.pt

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

checkpoint-12105/trainer_state.json

@@ -0,0 +1,211 @@
+{
+  "best_metric": 0.5624491572380066,
+  "best_model_checkpoint": "galaxy_type_image_detection/checkpoint-12105",
+  "epoch": 1.0,
+  "eval_steps": 500,
+  "global_step": 12105,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.041305245766212306,
+      "grad_norm": 1.0393656492233276,
+      "learning_rate": 8.66403981750311e-07,
+      "loss": 0.967,
+      "step": 500
+    },
+    {
+      "epoch": 0.08261049153242461,
+      "grad_norm": 2.2510411739349365,
+      "learning_rate": 8.2907507258399e-07,
+      "loss": 0.8018,
+      "step": 1000
+    },
+    {
+      "epoch": 0.12391573729863693,
+      "grad_norm": 2.212888479232788,
+      "learning_rate": 7.91746163417669e-07,
+      "loss": 0.7377,
+      "step": 1500
+    },
+    {
+      "epoch": 0.16522098306484923,
+      "grad_norm": 2.6769936084747314,
+      "learning_rate": 7.54417254251348e-07,
+      "loss": 0.6944,
+      "step": 2000
+    },
+    {
+      "epoch": 0.20652622883106153,
+      "grad_norm": 1.5611211061477661,
+      "learning_rate": 7.170883450850269e-07,
+      "loss": 0.6649,
+      "step": 2500
+    },
+    {
+      "epoch": 0.24783147459727387,
+      "grad_norm": 2.753368616104126,
+      "learning_rate": 6.797594359187059e-07,
+      "loss": 0.6495,
+      "step": 3000
+    },
+    {
+      "epoch": 0.28913672036348614,
+      "grad_norm": 2.5022592544555664,
+      "learning_rate": 6.424305267523848e-07,
+      "loss": 0.6282,
+      "step": 3500
+    },
+    {
+      "epoch": 0.33044196612969845,
+      "grad_norm": 3.5701966285705566,
+      "learning_rate": 6.051016175860639e-07,
+      "loss": 0.6171,
+      "step": 4000
+    },
+    {
+      "epoch": 0.37174721189591076,
+      "grad_norm": 2.316419839859009,
+      "learning_rate": 5.677727084197428e-07,
+      "loss": 0.609,
+      "step": 4500
+    },
+    {
+      "epoch": 0.41305245766212306,
+      "grad_norm": 2.764505624771118,
+      "learning_rate": 5.304437992534219e-07,
+      "loss": 0.5999,
+      "step": 5000
+    },
+    {
+      "epoch": 0.4543577034283354,
+      "grad_norm": 3.0616867542266846,
+      "learning_rate": 4.931148900871008e-07,
+      "loss": 0.5925,
+      "step": 5500
+    },
+    {
+      "epoch": 0.49566294919454773,
+      "grad_norm": 3.7850115299224854,
+      "learning_rate": 4.557859809207797e-07,
+      "loss": 0.589,
+      "step": 6000
+    },
+    {
+      "epoch": 0.53696819496076,
+      "grad_norm": 2.5644021034240723,
+      "learning_rate": 4.1845707175445867e-07,
+      "loss": 0.5829,
+      "step": 6500
+    },
+    {
+      "epoch": 0.5782734407269723,
+      "grad_norm": 2.6693475246429443,
+      "learning_rate": 3.8112816258813767e-07,
+      "loss": 0.5768,
+      "step": 7000
+    },
+    {
+      "epoch": 0.6195786864931846,
+      "grad_norm": 4.514222145080566,
+      "learning_rate": 3.437992534218167e-07,
+      "loss": 0.5728,
+      "step": 7500
+    },
+    {
+      "epoch": 0.6608839322593969,
+      "grad_norm": 3.1888978481292725,
+      "learning_rate": 3.064703442554956e-07,
+      "loss": 0.5698,
+      "step": 8000
+    },
+    {
+      "epoch": 0.7021891780256092,
+      "grad_norm": 3.4488654136657715,
+      "learning_rate": 2.691414350891746e-07,
+      "loss": 0.5737,
+      "step": 8500
+    },
+    {
+      "epoch": 0.7434944237918215,
+      "grad_norm": 4.471464157104492,
+      "learning_rate": 2.3181252592285358e-07,
+      "loss": 0.5643,
+      "step": 9000
+    },
+    {
+      "epoch": 0.7847996695580338,
+      "grad_norm": 3.535975217819214,
+      "learning_rate": 1.9448361675653255e-07,
+      "loss": 0.5682,
+      "step": 9500
+    },
+    {
+      "epoch": 0.8261049153242461,
+      "grad_norm": 5.046966075897217,
+      "learning_rate": 1.571547075902115e-07,
+      "loss": 0.5628,
+      "step": 10000
+    },
+    {
+      "epoch": 0.8674101610904585,
+      "grad_norm": 3.240630626678467,
+      "learning_rate": 1.198257984238905e-07,
+      "loss": 0.5613,
+      "step": 10500
+    },
+    {
+      "epoch": 0.9087154068566708,
+      "grad_norm": 4.973729133605957,
+      "learning_rate": 8.249688925756947e-08,
+      "loss": 0.5602,
+      "step": 11000
+    },
+    {
+      "epoch": 0.9500206526228832,
+      "grad_norm": 4.114513397216797,
+      "learning_rate": 4.516798009124844e-08,
+      "loss": 0.5622,
+      "step": 11500
+    },
+    {
+      "epoch": 0.9913258983890955,
+      "grad_norm": 3.011930227279663,
+      "learning_rate": 7.839070924927416e-09,
+      "loss": 0.5592,
+      "step": 12000
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7654878108598617,
+      "eval_loss": 0.5624491572380066,
+      "eval_model_preparation_time": 0.006,
+      "eval_runtime": 413.2412,
+      "eval_samples_per_second": 98.669,
+      "eval_steps_per_second": 3.085,
+      "step": 12105
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 12105,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 6.003272999208671e+19,
+  "train_batch_size": 64,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-12105/training_args.bin

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

config.json

@@ -9,17 +9,17 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "S",
-    "1": "SB",
-    "2": "E"
+    "0": "E",
+    "1": "S",
+    "2": "SB"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "E": 2,
-    "S": 0,
-    "SB": 1
+    "E": 0,
+    "S": 1,
+    "SB": 2
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
@@ -30,5 +30,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.32.1"
+  "transformers_version": "4.44.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:565bb0644333a10c5d28c3cf40111f193c66d72510527072d071b50f07adca81
+oid sha256:b33b2b1054d985b32e22181fbf4e011685b83317ec0dac8a1ee688ac78a0142c
 size 343227052

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:84330c3b6e1bac0306f601121d873a8d06d7a1eac19d3e58de78cde0aa8316db
-size 4027
+oid sha256:b6dc03e977829a9fe1632bfde9dc8286b9b4d9e269bca46e17c31045d8f211d2
+size 4667