Upload FIM-ODE fhn model

fhn/gridsearch_01-29-1048/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": 5,
+    "loss_filter_nans": true,
+    "loss_type": "l1",
+    "num_ic": 1,
+    "only_final_points_for_loss": false,
+    "step_noise_scale": 0.0,
+    "train_type": "trajectory_reconstruction",
+    "train_with_normalized_head": true,
+    "traj_loss_steps": 19,
+    "use_h_max": false
+  },
+  "transformers_version": "4.46.0"
+}

fhn/gridsearch_01-29-1048/checkpoints/best-model/model.safetensors

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

fhn/gridsearch_01-29-1048/checkpoints/best-model/optimizers-checkpoint.pth

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

fhn/gridsearch_01-29-1048/checkpoints/best-model/train-state-checkpoint.pth

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