End of training

README.md

@@ -1,7 +1,6 @@
 ---
 library_name: transformers
 tags:
-- image-classification
 - generated_from_trainer
 model-index:
 - name: vae_test
@@ -13,7 +12,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vae_test
 
-This model is a fine-tuned version of [](https://huggingface.co/) on the train_file=/home/pj24002027/ku50001104/data/mutual_dataset/few_data/train.jsonl, validation_file=/home/pj24002027/ku50001104/data/mutual_dataset/few_data/test.jsonl, max_train_samples=2048, max_eval_samples=2048, use_sensor_keys=CAM_FRONT dataset.
+This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
 
 ## Model description
 

all_results.json

@@ -1,7 +1,7 @@
 {
     "epoch": 0.125,
-    "train_loss": 1.1237460374832153,
-    "train_runtime": 20.6388,
-    "train_samples_per_second": 9.923,
-    "train_steps_per_second": 0.097
+    "train_loss": 1.3442257046699524,
+    "train_runtime": 21.7074,
+    "train_samples_per_second": 9.435,
+    "train_steps_per_second": 0.092
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d3385c46b833dccf98c5e3ceb6b4a9b174795a85374120706fa0a0c9a42f2197
+oid sha256:b58258e12f0e3503a9ac6d19f156a0ebb77b4e282f1d09e43335f3a5d7939966
 size 31338740

my_config.json

@@ -1,6 +1,6 @@
 {
     "output_dir": "logs/vae_test",
-    "overwrite_output_dir": true,
+    "overwrite_output_dir": false,
     "model_type": "vae",
     "report_to": [
         "wandb"

train_results.json

@@ -1,7 +1,7 @@
 {
     "epoch": 0.125,
-    "train_loss": 1.1237460374832153,
-    "train_runtime": 20.6388,
-    "train_samples_per_second": 9.923,
-    "train_steps_per_second": 0.097
+    "train_loss": 1.3442257046699524,
+    "train_runtime": 21.7074,
+    "train_samples_per_second": 9.435,
+    "train_steps_per_second": 0.092
 }

trainer_state.json

@@ -11,26 +11,26 @@
   "log_history": [
     {
       "epoch": 0.0625,
-      "grad_norm": 4.648393154144287,
+      "grad_norm": 6.13511848449707,
       "learning_rate": 0.0,
-      "loss": 1.1228,
+      "loss": 1.3392,
       "step": 1
     },
     {
       "epoch": 0.125,
-      "grad_norm": 4.56954288482666,
+      "grad_norm": 6.1215314865112305,
       "learning_rate": 2.0000000000000002e-07,
-      "loss": 1.1247,
+      "loss": 1.3493,
       "step": 2
     },
     {
       "epoch": 0.125,
       "step": 2,
       "total_flos": 147645328785408.0,
-      "train_loss": 1.1237460374832153,
-      "train_runtime": 20.6388,
-      "train_samples_per_second": 9.923,
-      "train_steps_per_second": 0.097
+      "train_loss": 1.3442257046699524,
+      "train_runtime": 21.7074,
+      "train_samples_per_second": 9.435,
+      "train_steps_per_second": 0.092
     }
   ],
   "logging_steps": 1,

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:494f49e8ffe42f196e661babfa0f4516d40ccf2f9a923a613986c80ce7a70477
+oid sha256:5625939faf506c5952d731e03342453532cfdf8fa83e8ecc6feb4dba0dc45a5d
 size 5368