Epoch 25/30 | loss=0.0000

checkpoint-ep025/adapter_config.json

@@ -16,7 +16,7 @@
   "layers_pattern": null,
   "layers_to_transform": null,
   "loftq_config": {},
-  "lora_alpha": 16,
+  "lora_alpha": 32,
   "lora_bias": false,
   "lora_dropout": 0.1,
   "megatron_config": null,
@@ -25,11 +25,13 @@
   "peft_type": "LORA",
   "peft_version": "0.18.1",
   "qalora_group_size": 16,
-  "r": 8,
+  "r": 16,
   "rank_pattern": {},
   "revision": null,
   "target_modules": [
+    "o",
     "q",
+    "k",
     "v"
   ],
   "target_parameters": null,

checkpoint-ep025/adapter_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0616171dfc48585ac5af92efca4705e259d8a99c92034f6f7c350d77be686484
-size 3558888
+oid sha256:5092a75e0cb293d01a18feb5a08729e0e91e8062e4fa0eae8618389b46316428
+size 14196208

checkpoint-ep025/motion_encoder.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:aa348c075bdc42c380a895b44eb00a03f429a682c954d6453af5fb781c99cf52
-size 19092969
+oid sha256:1ffa97be298ef5067e2c79ac576801ef5757268e91ccdc46759cd7423b5d2d75
+size 37689097

checkpoint-ep025/slt_config.json

@@ -1,11 +1,13 @@
 {
+  "pose_dim": 182,
   "codebook_dim": 512,
   "enc_hidden": 256,
-  "enc_layers": 4,
+  "enc_layers": 6,
   "enc_heads": 8,
-  "enc_ff": 1024,
+  "enc_ff": 2048,
   "enc_dropout": 0.1,
   "t5_hidden": 768,
   "base_model": "google/flan-t5-base",
-  "model_type": "m2t_slt"
+  "model_type": "m2t_slt",
+  "has_vqvae_decoder": true
 }

checkpoint-ep025/vqvae_decoder.pt

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