End of training

README.md

@@ -17,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [MCG-NJU/videomae-base](https://huggingface.co/MCG-NJU/videomae-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.7432
+- Loss: 0.7424
 - F1: 0.5798
 
 ## Model description

all_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 9.06,
+    "eval_f1": 0.5797752808988764,
+    "eval_loss": 0.7424257397651672,
+    "eval_runtime": 135.4113,
+    "eval_samples_per_second": 2.814,
+    "eval_steps_per_second": 0.177
+}

test_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 9.06,
+    "eval_f1": 0.5797752808988764,
+    "eval_loss": 0.7424257397651672,
+    "eval_runtime": 135.4113,
+    "eval_samples_per_second": 2.814,
+    "eval_steps_per_second": 0.177
+}

trainer_state.json

@@ -0,0 +1,306 @@
+{
+  "best_metric": 0.49748743718592964,
+  "best_model_checkpoint": "videomae-base-finetuned-ElderReact-anger-balancedf1/checkpoint-75",
+  "epoch": 9.0625,
+  "eval_steps": 500,
+  "global_step": 240,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.04,
+      "grad_norm": 3.997068166732788,
+      "learning_rate": 2.0833333333333336e-05,
+      "loss": 0.7186,
+      "step": 10
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 3.005845785140991,
+      "learning_rate": 4.166666666666667e-05,
+      "loss": 0.7339,
+      "step": 20
+    },
+    {
+      "epoch": 0.1,
+      "eval_f1": 0.4463276836158192,
+      "eval_loss": 0.6864726543426514,
+      "eval_runtime": 171.9816,
+      "eval_samples_per_second": 2.442,
+      "eval_steps_per_second": 0.157,
+      "step": 25
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 6.235264301300049,
+      "learning_rate": 4.8611111111111115e-05,
+      "loss": 0.7253,
+      "step": 30
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 3.536816358566284,
+      "learning_rate": 4.62962962962963e-05,
+      "loss": 0.7024,
+      "step": 40
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 12.325151443481445,
+      "learning_rate": 4.3981481481481486e-05,
+      "loss": 0.7007,
+      "step": 50
+    },
+    {
+      "epoch": 1.1,
+      "eval_f1": 0.047619047619047616,
+      "eval_loss": 0.7107335925102234,
+      "eval_runtime": 176.4951,
+      "eval_samples_per_second": 2.38,
+      "eval_steps_per_second": 0.153,
+      "step": 50
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 2.6023736000061035,
+      "learning_rate": 4.166666666666667e-05,
+      "loss": 0.6583,
+      "step": 60
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 4.512753963470459,
+      "learning_rate": 3.935185185185186e-05,
+      "loss": 0.6854,
+      "step": 70
+    },
+    {
+      "epoch": 2.1,
+      "eval_f1": 0.49748743718592964,
+      "eval_loss": 0.6994316577911377,
+      "eval_runtime": 178.8944,
+      "eval_samples_per_second": 2.348,
+      "eval_steps_per_second": 0.151,
+      "step": 75
+    },
+    {
+      "epoch": 3.02,
+      "grad_norm": 5.125423908233643,
+      "learning_rate": 3.7037037037037037e-05,
+      "loss": 0.7021,
+      "step": 80
+    },
+    {
+      "epoch": 3.06,
+      "grad_norm": 3.5603673458099365,
+      "learning_rate": 3.472222222222222e-05,
+      "loss": 0.6419,
+      "step": 90
+    },
+    {
+      "epoch": 3.1,
+      "grad_norm": 21.498886108398438,
+      "learning_rate": 3.240740740740741e-05,
+      "loss": 0.715,
+      "step": 100
+    },
+    {
+      "epoch": 3.1,
+      "eval_f1": 0.13333333333333333,
+      "eval_loss": 0.7238408923149109,
+      "eval_runtime": 178.3055,
+      "eval_samples_per_second": 2.356,
+      "eval_steps_per_second": 0.151,
+      "step": 100
+    },
+    {
+      "epoch": 4.04,
+      "grad_norm": 3.3730030059814453,
+      "learning_rate": 3.0092592592592593e-05,
+      "loss": 0.6555,
+      "step": 110
+    },
+    {
+      "epoch": 4.08,
+      "grad_norm": 6.0418829917907715,
+      "learning_rate": 2.777777777777778e-05,
+      "loss": 0.6866,
+      "step": 120
+    },
+    {
+      "epoch": 4.1,
+      "eval_f1": 0.01951219512195122,
+      "eval_loss": 0.7563002109527588,
+      "eval_runtime": 176.1376,
+      "eval_samples_per_second": 2.384,
+      "eval_steps_per_second": 0.153,
+      "step": 125
+    },
+    {
+      "epoch": 5.02,
+      "grad_norm": 6.097543716430664,
+      "learning_rate": 2.5462962962962965e-05,
+      "loss": 0.7311,
+      "step": 130
+    },
+    {
+      "epoch": 5.06,
+      "grad_norm": 2.698549509048462,
+      "learning_rate": 2.314814814814815e-05,
+      "loss": 0.5913,
+      "step": 140
+    },
+    {
+      "epoch": 5.1,
+      "grad_norm": 17.406471252441406,
+      "learning_rate": 2.0833333333333336e-05,
+      "loss": 0.6564,
+      "step": 150
+    },
+    {
+      "epoch": 5.1,
+      "eval_f1": 0.15637860082304528,
+      "eval_loss": 0.7628450393676758,
+      "eval_runtime": 177.1888,
+      "eval_samples_per_second": 2.37,
+      "eval_steps_per_second": 0.152,
+      "step": 150
+    },
+    {
+      "epoch": 6.04,
+      "grad_norm": 4.0975470542907715,
+      "learning_rate": 1.8518518518518518e-05,
+      "loss": 0.598,
+      "step": 160
+    },
+    {
+      "epoch": 6.08,
+      "grad_norm": 4.26428747177124,
+      "learning_rate": 1.6203703703703704e-05,
+      "loss": 0.5809,
+      "step": 170
+    },
+    {
+      "epoch": 6.1,
+      "eval_f1": 0.38311688311688313,
+      "eval_loss": 0.7543146014213562,
+      "eval_runtime": 167.9465,
+      "eval_samples_per_second": 2.501,
+      "eval_steps_per_second": 0.161,
+      "step": 175
+    },
+    {
+      "epoch": 7.02,
+      "grad_norm": 2.8742892742156982,
+      "learning_rate": 1.388888888888889e-05,
+      "loss": 0.4884,
+      "step": 180
+    },
+    {
+      "epoch": 7.06,
+      "grad_norm": 4.310091972351074,
+      "learning_rate": 1.1574074074074075e-05,
+      "loss": 0.5761,
+      "step": 190
+    },
+    {
+      "epoch": 7.1,
+      "grad_norm": 3.7042274475097656,
+      "learning_rate": 9.259259259259259e-06,
+      "loss": 0.6035,
+      "step": 200
+    },
+    {
+      "epoch": 7.1,
+      "eval_f1": 0.21224489795918366,
+      "eval_loss": 0.7982825040817261,
+      "eval_runtime": 168.7315,
+      "eval_samples_per_second": 2.489,
+      "eval_steps_per_second": 0.16,
+      "step": 200
+    },
+    {
+      "epoch": 8.04,
+      "grad_norm": 4.604009628295898,
+      "learning_rate": 6.944444444444445e-06,
+      "loss": 0.5098,
+      "step": 210
+    },
+    {
+      "epoch": 8.08,
+      "grad_norm": 3.782595157623291,
+      "learning_rate": 4.6296296296296296e-06,
+      "loss": 0.6153,
+      "step": 220
+    },
+    {
+      "epoch": 8.1,
+      "eval_f1": 0.40634920634920635,
+      "eval_loss": 0.7542010545730591,
+      "eval_runtime": 176.9739,
+      "eval_samples_per_second": 2.373,
+      "eval_steps_per_second": 0.153,
+      "step": 225
+    },
+    {
+      "epoch": 9.02,
+      "grad_norm": 3.7313973903656006,
+      "learning_rate": 2.3148148148148148e-06,
+      "loss": 0.5072,
+      "step": 230
+    },
+    {
+      "epoch": 9.06,
+      "grad_norm": 2.534163475036621,
+      "learning_rate": 0.0,
+      "loss": 0.5099,
+      "step": 240
+    },
+    {
+      "epoch": 9.06,
+      "eval_f1": 0.4525993883792049,
+      "eval_loss": 0.7408385276794434,
+      "eval_runtime": 159.3385,
+      "eval_samples_per_second": 2.636,
+      "eval_steps_per_second": 0.169,
+      "step": 240
+    },
+    {
+      "epoch": 9.06,
+      "step": 240,
+      "total_flos": 4.616669303992812e+18,
+      "train_loss": 0.6372330923875172,
+      "train_runtime": 3422.175,
+      "train_samples_per_second": 1.122,
+      "train_steps_per_second": 0.07
+    },
+    {
+      "epoch": 9.06,
+      "eval_f1": 0.5797752808988764,
+      "eval_loss": 0.7431801557540894,
+      "eval_runtime": 150.1599,
+      "eval_samples_per_second": 2.537,
+      "eval_steps_per_second": 0.16,
+      "step": 240
+    },
+    {
+      "epoch": 9.06,
+      "eval_f1": 0.5797752808988764,
+      "eval_loss": 0.7424257397651672,
+      "eval_runtime": 135.4113,
+      "eval_samples_per_second": 2.814,
+      "eval_steps_per_second": 0.177,
+      "step": 240
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 240,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 9223372036854775807,
+  "save_steps": 500,
+  "total_flos": 4.616669303992812e+18,
+  "train_batch_size": 16,
+  "trial_name": null,
+  "trial_params": null
+}