jrhuebers commited on
Commit
b65fa6c
·
verified ·
1 Parent(s): b8f4b1f

Upload FIM-ODE mocap09long model

Browse files
mocap09long/mocap09long_01-29-0609/checkpoints/best-model/config.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "TrainingWrapper"
4
+ ],
5
+ "model_config": {
6
+ "attention_map": "softmax",
7
+ "attention_method": "linear",
8
+ "dim_embed": 256,
9
+ "dim_feedforward": 1024,
10
+ "dim_ffn_u_model": 1024,
11
+ "dim_hidden_u_model": 256,
12
+ "dim_max_trajectory": 3,
13
+ "dropout": 0.0,
14
+ "num_context_encoder_layers": 2,
15
+ "num_heads": 8,
16
+ "num_res_layer_u_model": 6,
17
+ "num_res_layers_functional_decoder": 8,
18
+ "use_bias_for_projection": true,
19
+ "use_bias_in_attention": true,
20
+ "use_query_residual_in_attention": true
21
+ },
22
+ "torch_dtype": "float32",
23
+ "train_config": {
24
+ "corruption_model_type": null,
25
+ "h_max": null,
26
+ "ic_noise_scale": 0.0,
27
+ "integrator_for_trajectory_training": "rk4",
28
+ "intermediate_steps_per_step": 1,
29
+ "loss_filter_nans": true,
30
+ "loss_type": "l1",
31
+ "num_ic": 7,
32
+ "random_initial_conditions": false,
33
+ "step_noise_scale": 0.2,
34
+ "train_type": "trajectory_reconstruction",
35
+ "train_with_normalized_head": true,
36
+ "traj_loss_steps": 25,
37
+ "use_h_max": false
38
+ },
39
+ "transformers_version": "4.46.0"
40
+ }
mocap09long/mocap09long_01-29-0609/checkpoints/best-model/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:41ec8e7511bb6b2cd5aa283ddce17ad0da34501311847d1d389722106dce26ff
3
+ size 51907416
mocap09long/mocap09long_01-29-0609/checkpoints/best-model/optimizers-checkpoint.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:78ade42ecbb13f95445b0f4f0c0abee36f438bc7708bf94c06d4dc1c9f9d3405
3
+ size 64937563
mocap09long/mocap09long_01-29-0609/checkpoints/best-model/train-state-checkpoint.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:805d0fd50d6b842967f81989045a1d98473aebc688a35fcafc2d730b25543466
3
+ size 848449