Upload FIM-ODE vdp1 model

vdp1/vdp1_01-29-0707/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": "l1",
+    "num_ic": 3,
+    "random_initial_conditions": false,
+    "step_noise_scale": 0.0,
+    "train_type": "trajectory_reconstruction",
+    "train_with_normalized_head": true,
+    "traj_loss_steps": 25,
+    "use_h_max": false
+  },
+  "transformers_version": "4.46.0"
+}

vdp1/vdp1_01-29-0707/checkpoints/best-model/model.safetensors

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

vdp1/vdp1_01-29-0707/checkpoints/best-model/optimizers-checkpoint.pth

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

vdp1/vdp1_01-29-0707/checkpoints/best-model/train-state-checkpoint.pth

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