Upload FIM-ODE mocap39long model

mocap39long/mocap39long_01-29-0619/checkpoints/best-model/config.json

@@ -0,0 +1,40 @@
+{
+  "architectures": [
+    "TrainingWrapper"
+  ],
+  "model_config": {
+    "attention_map": "softmax",
+    "attention_method": "linear",
+    "dim_embed": 256,
+    "dim_feedforward": 1024,
+    "dim_ffn_u_model": 1024,
+    "dim_hidden_u_model": 256,
+    "dim_max_trajectory": 3,
+    "dropout": 0.0,
+    "num_context_encoder_layers": 2,
+    "num_heads": 8,
+    "num_res_layer_u_model": 6,
+    "num_res_layers_functional_decoder": 8,
+    "use_bias_for_projection": true,
+    "use_bias_in_attention": true,
+    "use_query_residual_in_attention": true
+  },
+  "torch_dtype": "float32",
+  "train_config": {
+    "corruption_model_type": null,
+    "h_max": null,
+    "ic_noise_scale": 0.0,
+    "integrator_for_trajectory_training": "rk4",
+    "intermediate_steps_per_step": 1,
+    "loss_filter_nans": true,
+    "loss_type": "mse",
+    "num_ic": 19,
+    "random_initial_conditions": false,
+    "step_noise_scale": 0.2,
+    "train_type": "trajectory_reconstruction",
+    "train_with_normalized_head": true,
+    "traj_loss_steps": 25,
+    "use_h_max": false
+  },
+  "transformers_version": "4.46.0"
+}

mocap39long/mocap39long_01-29-0619/checkpoints/best-model/model.safetensors

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

mocap39long/mocap39long_01-29-0619/checkpoints/best-model/optimizers-checkpoint.pth

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

mocap39long/mocap39long_01-29-0619/checkpoints/best-model/train-state-checkpoint.pth

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