Upload the ETH_UCY and SDD checkpoints

eth_ucy/moflow/eth/checkpoint_best.pt

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

eth_ucy/moflow/eth/cor_fm.yml

@@ -0,0 +1,93 @@
+LOSS_NN_MODE: agent
+LOSS_REG_REDUCTION: sum
+LOSS_REG_SQUARED: false
+LOSS_VELOCITY: false
+MODEL:
+  CLASSIFICATION_MLPS:
+  - 128
+  - 128
+  - 1
+  CONTEXT_ENCODER:
+    AGENTS: 2
+    DROPOUT_OF_ATTN: 0.1
+    D_MODEL: 128
+    NAME: ETHEncoder
+    NUM_ATTN_HEAD: 8
+    NUM_ATTN_LAYERS: 4
+    NUM_CHANNEL_IN_MLP_AGENT: 256
+    NUM_INPUT_CONTEXT: 6
+    NUM_LAYER_IN_MLP_AGENT: 3
+  DETERMINISTIC: false
+  DROPOUT: null
+  MODEL_OUT_DIM: 24
+  MOTION_DECODER:
+    DROPOUT_OF_ATTN: 0.1
+    D_MODEL: 128
+    NAME: MTRDecoder
+    NUM_ATTN_HEAD: 8
+    NUM_DECODER_BLOCKS: 4
+  NUM_LAYERS: null
+  NUM_PROPOSED_QUERY: 20
+  REGRESSION_MLPS:
+  - 128
+  - 256
+  - 24
+  USE_PRE_NORM: false
+OPTIMIZATION:
+  GRAD_NORM_CLIP: 1.0
+  LOSS_WEIGHTS:
+    cls: 1.0
+    reg: 1.0
+    vel: 0.2
+  LR: 0.0003
+  LR_CLIP: 1.0e-06
+  NUM_EPOCHS: 150
+  OPTIMIZER: AdamW
+  SCHEDULER: cosineAnnealingLRwithWarmup
+  WEIGHT_DECAY: 0.01
+ROOT: /bigdata/fyx/BehaviorModelling
+agents: 2
+cfg_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_eth_rot_6_min_max_LR0.0003_WD0.01_CLS_1.0_BS32_EP150
+cfg_name: cor_fm
+cfg_path: cfg/eth_ucy/cor_fm.yml
+checkpt_freq: 1
+data_norm: min_max
+dataset: eth_ucy
+denoising_head_preds: 20
+denoising_method: fm
+description: Correlated FlowMatching
+device: cuda
+drop_logi_k: 20.0
+drop_logi_m: 0.5
+drop_method: emb
+fm_in_scaling: true
+fm_rew_sqrt: false
+fm_wrapper: direct
+future_frames: 12
+k_preds: 20
+log_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_eth_rot_6_min_max_LR0.0003_WD0.01_CLS_1.0_BS32_EP150/log
+logit_norm_mean: -0.5
+logit_norm_std: 1.5
+max_num_ckpts: 5
+model_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_eth_rot_6_min_max_LR0.0003_WD0.01_CLS_1.0_BS32_EP150/models
+model_files: []
+model_path: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_eth_rot_6_min_max_LR0.0003_WD0.01_CLS_1.0_BS32_EP150/models/model_%04d.p
+notes: Flow Matching
+objective: pred_data
+past_frames: 8
+plot_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_eth_rot_6_min_max_LR0.0003_WD0.01_CLS_1.0_BS32_EP150/plots
+results_root_dir: results_eth_ucy
+rotate: true
+rotate_aug: false
+rotate_time_frame: 6
+sample_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_eth_rot_6_min_max_LR0.0003_WD0.01_CLS_1.0_BS32_EP150/samples
+sampling_steps: 10
+sigma_data: 0.13
+subset: eth
+t_schedule: logit_normal
+tag: rjliao-01.ece.ubc.ca_v0.05
+test_batch_size: 64
+tied_noise: true
+train_batch_size: 32
+train_mode: true
+

eth_ucy/moflow/hotel/checkpoint_best.pt

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

eth_ucy/moflow/hotel/cor_fm.yml

@@ -0,0 +1,93 @@
+LOSS_NN_MODE: agent
+LOSS_REG_REDUCTION: sum
+LOSS_REG_SQUARED: false
+LOSS_VELOCITY: false
+MODEL:
+  CLASSIFICATION_MLPS:
+  - 128
+  - 128
+  - 1
+  CONTEXT_ENCODER:
+    AGENTS: 2
+    DROPOUT_OF_ATTN: 0.1
+    D_MODEL: 128
+    NAME: ETHEncoder
+    NUM_ATTN_HEAD: 8
+    NUM_ATTN_LAYERS: 4
+    NUM_CHANNEL_IN_MLP_AGENT: 256
+    NUM_INPUT_CONTEXT: 6
+    NUM_LAYER_IN_MLP_AGENT: 3
+  DETERMINISTIC: false
+  DROPOUT: null
+  MODEL_OUT_DIM: 24
+  MOTION_DECODER:
+    DROPOUT_OF_ATTN: 0.1
+    D_MODEL: 128
+    NAME: MTRDecoder
+    NUM_ATTN_HEAD: 8
+    NUM_DECODER_BLOCKS: 4
+  NUM_LAYERS: null
+  NUM_PROPOSED_QUERY: 20
+  REGRESSION_MLPS:
+  - 128
+  - 256
+  - 24
+  USE_PRE_NORM: false
+OPTIMIZATION:
+  GRAD_NORM_CLIP: 1.0
+  LOSS_WEIGHTS:
+    cls: 1.0
+    reg: 1.0
+    vel: 0.2
+  LR: 0.0005
+  LR_CLIP: 1.0e-06
+  NUM_EPOCHS: 150
+  OPTIMIZER: AdamW
+  SCHEDULER: cosineAnnealingLRwithWarmup
+  WEIGHT_DECAY: 0.01
+ROOT: /bigdata/fyx/BehaviorModelling
+agents: 2
+cfg_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_hotel_rot_6_min_max_LR0.0005_WD0.01_CLS_1.0_BS48_EP150
+cfg_name: cor_fm
+cfg_path: cfg/eth_ucy/cor_fm.yml
+checkpt_freq: 1
+data_norm: min_max
+dataset: eth_ucy
+denoising_head_preds: 20
+denoising_method: fm
+description: Correlated FlowMatching
+device: cuda
+drop_logi_k: 20.0
+drop_logi_m: 0.5
+drop_method: emb
+fm_in_scaling: true
+fm_rew_sqrt: false
+fm_wrapper: direct
+future_frames: 12
+k_preds: 20
+log_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_hotel_rot_6_min_max_LR0.0005_WD0.01_CLS_1.0_BS48_EP150/log
+logit_norm_mean: -0.5
+logit_norm_std: 1.5
+max_num_ckpts: 5
+model_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_hotel_rot_6_min_max_LR0.0005_WD0.01_CLS_1.0_BS48_EP150/models
+model_files: []
+model_path: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_hotel_rot_6_min_max_LR0.0005_WD0.01_CLS_1.0_BS48_EP150/models/model_%04d.p
+notes: Flow Matching
+objective: pred_data
+past_frames: 8
+plot_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_hotel_rot_6_min_max_LR0.0005_WD0.01_CLS_1.0_BS48_EP150/plots
+results_root_dir: results_eth_ucy
+rotate: true
+rotate_aug: false
+rotate_time_frame: 6
+sample_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_hotel_rot_6_min_max_LR0.0005_WD0.01_CLS_1.0_BS48_EP150/samples
+sampling_steps: 10
+sigma_data: 0.13
+subset: hotel
+t_schedule: logit_normal
+tag: rjliao-01.ece.ubc.ca_v0.05
+test_batch_size: 96
+tied_noise: true
+train_batch_size: 48
+train_mode: true
+

eth_ucy/moflow/univ/checkpoint_best.pt

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

eth_ucy/moflow/univ/cor_fm.yml

@@ -0,0 +1,93 @@
+LOSS_NN_MODE: agent
+LOSS_REG_REDUCTION: sum
+LOSS_REG_SQUARED: false
+LOSS_VELOCITY: false
+MODEL:
+  CLASSIFICATION_MLPS:
+  - 128
+  - 128
+  - 1
+  CONTEXT_ENCODER:
+    AGENTS: 2
+    DROPOUT_OF_ATTN: 0.1
+    D_MODEL: 128
+    NAME: ETHEncoder
+    NUM_ATTN_HEAD: 8
+    NUM_ATTN_LAYERS: 4
+    NUM_CHANNEL_IN_MLP_AGENT: 256
+    NUM_INPUT_CONTEXT: 6
+    NUM_LAYER_IN_MLP_AGENT: 3
+  DETERMINISTIC: false
+  DROPOUT: null
+  MODEL_OUT_DIM: 24
+  MOTION_DECODER:
+    DROPOUT_OF_ATTN: 0.1
+    D_MODEL: 128
+    NAME: MTRDecoder
+    NUM_ATTN_HEAD: 8
+    NUM_DECODER_BLOCKS: 4
+  NUM_LAYERS: null
+  NUM_PROPOSED_QUERY: 20
+  REGRESSION_MLPS:
+  - 128
+  - 256
+  - 24
+  USE_PRE_NORM: false
+OPTIMIZATION:
+  GRAD_NORM_CLIP: 1.0
+  LOSS_WEIGHTS:
+    cls: 1.0
+    reg: 1.0
+    vel: 0.2
+  LR: 0.0005
+  LR_CLIP: 1.0e-06
+  NUM_EPOCHS: 150
+  OPTIMIZER: AdamW
+  SCHEDULER: cosineAnnealingLRwithWarmup
+  WEIGHT_DECAY: 0.01
+ROOT: /bigdata/fyx/BehaviorModelling
+agents: 2
+cfg_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_univ_rot_6_min_max_LR0.0005_WD0.01_CLS_1.0_BS48_EP150
+cfg_name: cor_fm
+cfg_path: cfg/eth_ucy/cor_fm.yml
+checkpt_freq: 1
+data_norm: min_max
+dataset: eth_ucy
+denoising_head_preds: 20
+denoising_method: fm
+description: Correlated FlowMatching
+device: cuda
+drop_logi_k: 20.0
+drop_logi_m: 0.5
+drop_method: emb
+fm_in_scaling: true
+fm_rew_sqrt: false
+fm_wrapper: direct
+future_frames: 12
+k_preds: 20
+log_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_univ_rot_6_min_max_LR0.0005_WD0.01_CLS_1.0_BS48_EP150/log
+logit_norm_mean: -0.5
+logit_norm_std: 1.5
+max_num_ckpts: 5
+model_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_univ_rot_6_min_max_LR0.0005_WD0.01_CLS_1.0_BS48_EP150/models
+model_files: []
+model_path: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_univ_rot_6_min_max_LR0.0005_WD0.01_CLS_1.0_BS48_EP150/models/model_%04d.p
+notes: Flow Matching
+objective: pred_data
+past_frames: 8
+plot_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_univ_rot_6_min_max_LR0.0005_WD0.01_CLS_1.0_BS48_EP150/plots
+results_root_dir: results_eth_ucy
+rotate: true
+rotate_aug: false
+rotate_time_frame: 6
+sample_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_univ_rot_6_min_max_LR0.0005_WD0.01_CLS_1.0_BS48_EP150/samples
+sampling_steps: 10
+sigma_data: 0.13
+subset: univ
+t_schedule: logit_normal
+tag: rjliao-01.ece.ubc.ca_v0.05
+test_batch_size: 96
+tied_noise: true
+train_batch_size: 48
+train_mode: true
+

eth_ucy/moflow/zara1/checkpoint_best.pt

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

eth_ucy/moflow/zara1/cor_fm.yml

@@ -0,0 +1,93 @@
+LOSS_NN_MODE: agent
+LOSS_REG_REDUCTION: sum
+LOSS_REG_SQUARED: false
+LOSS_VELOCITY: false
+MODEL:
+  CLASSIFICATION_MLPS:
+  - 128
+  - 128
+  - 1
+  CONTEXT_ENCODER:
+    AGENTS: 2
+    DROPOUT_OF_ATTN: 0.1
+    D_MODEL: 128
+    NAME: ETHEncoder
+    NUM_ATTN_HEAD: 8
+    NUM_ATTN_LAYERS: 4
+    NUM_CHANNEL_IN_MLP_AGENT: 256
+    NUM_INPUT_CONTEXT: 6
+    NUM_LAYER_IN_MLP_AGENT: 3
+  DETERMINISTIC: false
+  DROPOUT: null
+  MODEL_OUT_DIM: 24
+  MOTION_DECODER:
+    DROPOUT_OF_ATTN: 0.1
+    D_MODEL: 128
+    NAME: MTRDecoder
+    NUM_ATTN_HEAD: 8
+    NUM_DECODER_BLOCKS: 4
+  NUM_LAYERS: null
+  NUM_PROPOSED_QUERY: 20
+  REGRESSION_MLPS:
+  - 128
+  - 256
+  - 24
+  USE_PRE_NORM: false
+OPTIMIZATION:
+  GRAD_NORM_CLIP: 1.0
+  LOSS_WEIGHTS:
+    cls: 1.0
+    reg: 1.0
+    vel: 0.2
+  LR: 0.0003
+  LR_CLIP: 1.0e-06
+  NUM_EPOCHS: 150
+  OPTIMIZER: AdamW
+  SCHEDULER: cosineAnnealingLRwithWarmup
+  WEIGHT_DECAY: 0.01
+ROOT: /bigdata/fyx/BehaviorModelling
+agents: 2
+cfg_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_zara1_rot_6_min_max_LR0.0003_WD0.01_CLS_1.0_BS32_EP150
+cfg_name: cor_fm
+cfg_path: cfg/eth_ucy/cor_fm.yml
+checkpt_freq: 1
+data_norm: min_max
+dataset: eth_ucy
+denoising_head_preds: 20
+denoising_method: fm
+description: Correlated FlowMatching
+device: cuda
+drop_logi_k: 20.0
+drop_logi_m: 0.5
+drop_method: emb
+fm_in_scaling: true
+fm_rew_sqrt: false
+fm_wrapper: direct
+future_frames: 12
+k_preds: 20
+log_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_zara1_rot_6_min_max_LR0.0003_WD0.01_CLS_1.0_BS32_EP150/log
+logit_norm_mean: -0.5
+logit_norm_std: 1.5
+max_num_ckpts: 5
+model_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_zara1_rot_6_min_max_LR0.0003_WD0.01_CLS_1.0_BS32_EP150/models
+model_files: []
+model_path: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_zara1_rot_6_min_max_LR0.0003_WD0.01_CLS_1.0_BS32_EP150/models/model_%04d.p
+notes: Flow Matching
+objective: pred_data
+past_frames: 8
+plot_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_zara1_rot_6_min_max_LR0.0003_WD0.01_CLS_1.0_BS32_EP150/plots
+results_root_dir: results_eth_ucy
+rotate: true
+rotate_aug: false
+rotate_time_frame: 6
+sample_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_zara1_rot_6_min_max_LR0.0003_WD0.01_CLS_1.0_BS32_EP150/samples
+sampling_steps: 10
+sigma_data: 0.13
+subset: zara1
+t_schedule: logit_normal
+tag: rjliao-01.ece.ubc.ca_v0.05
+test_batch_size: 64
+tied_noise: true
+train_batch_size: 32
+train_mode: true
+

eth_ucy/moflow/zara2/checkpoint_best.pt

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

eth_ucy/moflow/zara2/cor_fm.yml

@@ -0,0 +1,93 @@
+LOSS_NN_MODE: agent
+LOSS_REG_REDUCTION: sum
+LOSS_REG_SQUARED: false
+LOSS_VELOCITY: false
+MODEL:
+  CLASSIFICATION_MLPS:
+  - 128
+  - 128
+  - 1
+  CONTEXT_ENCODER:
+    AGENTS: 2
+    DROPOUT_OF_ATTN: 0.1
+    D_MODEL: 128
+    NAME: ETHEncoder
+    NUM_ATTN_HEAD: 8
+    NUM_ATTN_LAYERS: 4
+    NUM_CHANNEL_IN_MLP_AGENT: 256
+    NUM_INPUT_CONTEXT: 6
+    NUM_LAYER_IN_MLP_AGENT: 3
+  DETERMINISTIC: false
+  DROPOUT: null
+  MODEL_OUT_DIM: 24
+  MOTION_DECODER:
+    DROPOUT_OF_ATTN: 0.1
+    D_MODEL: 128
+    NAME: MTRDecoder
+    NUM_ATTN_HEAD: 8
+    NUM_DECODER_BLOCKS: 4
+  NUM_LAYERS: null
+  NUM_PROPOSED_QUERY: 20
+  REGRESSION_MLPS:
+  - 128
+  - 256
+  - 24
+  USE_PRE_NORM: false
+OPTIMIZATION:
+  GRAD_NORM_CLIP: 1.0
+  LOSS_WEIGHTS:
+    cls: 1.0
+    reg: 1.0
+    vel: 0.2
+  LR: 0.0001
+  LR_CLIP: 1.0e-06
+  NUM_EPOCHS: 150
+  OPTIMIZER: AdamW
+  SCHEDULER: cosineAnnealingLRwithWarmup
+  WEIGHT_DECAY: 0.01
+ROOT: /bigdata/fyx/BehaviorModelling
+agents: 2
+cfg_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_zara2_rot_6_min_max_LR0.0001_WD0.01_CLS_1.0_BS32_EP150
+cfg_name: cor_fm
+cfg_path: cfg/eth_ucy/cor_fm.yml
+checkpt_freq: 1
+data_norm: min_max
+dataset: eth_ucy
+denoising_head_preds: 20
+denoising_method: fm
+description: Correlated FlowMatching
+device: cuda
+drop_logi_k: 20.0
+drop_logi_m: 0.5
+drop_method: emb
+fm_in_scaling: true
+fm_rew_sqrt: false
+fm_wrapper: direct
+future_frames: 12
+k_preds: 20
+log_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_zara2_rot_6_min_max_LR0.0001_WD0.01_CLS_1.0_BS32_EP150/log
+logit_norm_mean: -0.5
+logit_norm_std: 1.5
+max_num_ckpts: 5
+model_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_zara2_rot_6_min_max_LR0.0001_WD0.01_CLS_1.0_BS32_EP150/models
+model_files: []
+model_path: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_zara2_rot_6_min_max_LR0.0001_WD0.01_CLS_1.0_BS32_EP150/models/model_%04d.p
+notes: Flow Matching
+objective: pred_data
+past_frames: 8
+plot_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_zara2_rot_6_min_max_LR0.0001_WD0.01_CLS_1.0_BS32_EP150/plots
+results_root_dir: results_eth_ucy
+rotate: true
+rotate_aug: false
+rotate_time_frame: 6
+sample_dir: results_eth_ucy/cor_fm/rjliao-01.ece.ubc.ca_v0.05_FM_S10_log_m-0.5_s1.5_dire_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_zara2_rot_6_min_max_LR0.0001_WD0.01_CLS_1.0_BS32_EP150/samples
+sampling_steps: 10
+sigma_data: 0.13
+subset: zara2
+t_schedule: logit_normal
+tag: rjliao-01.ece.ubc.ca_v0.05
+test_batch_size: 64
+tied_noise: true
+train_batch_size: 32
+train_mode: true
+

sdd/moflow/checkpoint_best.pt

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

sdd/moflow/cor_fm.yml

@@ -0,0 +1,92 @@
+LOSS_NN_MODE: agent
+LOSS_REG_REDUCTION: sum
+LOSS_REG_SQUARED: false
+LOSS_VELOCITY: false
+MODEL:
+  CLASSIFICATION_MLPS:
+  - 128
+  - 128
+  - 1
+  CONTEXT_ENCODER:
+    AGENTS: 1
+    DROPOUT_OF_ATTN: 0.1
+    D_MODEL: 128
+    NAME: ETHEncoder
+    NUM_ATTN_HEAD: 8
+    NUM_ATTN_LAYERS: 4
+    NUM_CHANNEL_IN_MLP_AGENT: 256
+    NUM_INPUT_CONTEXT: 6
+    NUM_LAYER_IN_MLP_AGENT: 3
+  DETERMINISTIC: false
+  DROPOUT: null
+  MODEL_OUT_DIM: 24
+  MOTION_DECODER:
+    DROPOUT_OF_ATTN: 0.1
+    D_MODEL: 128
+    NAME: MTRDecoder
+    NUM_ATTN_HEAD: 8
+    NUM_DECODER_BLOCKS: 4
+  NUM_LAYERS: null
+  NUM_PROPOSED_QUERY: 20
+  REGRESSION_MLPS:
+  - 128
+  - 256
+  - 24
+  USE_PRE_NORM: false
+OPTIMIZATION:
+  GRAD_NORM_CLIP: 1.0
+  LOSS_WEIGHTS:
+    cls: 1.0
+    reg: 1.0
+    vel: 0.2
+  LR: 0.0001
+  LR_CLIP: 1.0e-06
+  NUM_EPOCHS: 150
+  OPTIMIZER: AdamW
+  SCHEDULER: cosineAnnealingLRwithWarmup
+  WEIGHT_DECAY: 0.01
+ROOT: /scratch/ssd004/scratch/yxfu/BehaviorModelling
+agents: 1
+cfg_dir: results_sdd/cor_fm/gpu157.cluster.local_v0.05_FM_S10_log_m-0.5_s1.5_dire_PC_0.01_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_rot_6_min_max_LR0.0001_WD0.01_CLS_1.0_BS64_EP150
+cfg_name: cor_fm
+cfg_path: cfg/sdd/cor_fm.yml
+checkpt_freq: 1
+data_norm: min_max
+dataset: sdd
+denoising_head_preds: 20
+denoising_method: fm
+description: Correlated FlowMatching
+device: cuda
+drop_logi_k: 20.0
+drop_logi_m: 0.5
+drop_method: emb
+fm_in_scaling: true
+fm_rew_sqrt: false
+fm_wrapper: direct
+future_frames: 12
+k_preds: 20
+log_dir: results_sdd/cor_fm/gpu157.cluster.local_v0.05_FM_S10_log_m-0.5_s1.5_dire_PC_0.01_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_rot_6_min_max_LR0.0001_WD0.01_CLS_1.0_BS64_EP150/log
+logit_norm_mean: -0.5
+logit_norm_std: 1.5
+max_num_ckpts: 5
+model_dir: results_sdd/cor_fm/gpu157.cluster.local_v0.05_FM_S10_log_m-0.5_s1.5_dire_PC_0.01_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_rot_6_min_max_LR0.0001_WD0.01_CLS_1.0_BS64_EP150/models
+model_files: []
+model_path: results_sdd/cor_fm/gpu157.cluster.local_v0.05_FM_S10_log_m-0.5_s1.5_dire_PC_0.01_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_rot_6_min_max_LR0.0001_WD0.01_CLS_1.0_BS64_EP150/models/model_%04d.p
+notes: Flow Matching
+objective: pred_data
+past_frames: 8
+plot_dir: results_sdd/cor_fm/gpu157.cluster.local_v0.05_FM_S10_log_m-0.5_s1.5_dire_PC_0.01_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_rot_6_min_max_LR0.0001_WD0.01_CLS_1.0_BS64_EP150/plots
+results_root_dir: results_sdd
+rotate: true
+rotate_aug: false
+rotate_time_frame: 6
+sample_dir: results_sdd/cor_fm/gpu157.cluster.local_v0.05_FM_S10_log_m-0.5_s1.5_dire_PC_0.01_drop_emb_m0.5_k20.0_IS_TN_NN_A_REG_S_rot_6_min_max_LR0.0001_WD0.01_CLS_1.0_BS64_EP150/samples
+sampling_steps: 10
+sigma_data: 0.0663
+t_schedule: logit_normal
+tag: gpu157.cluster.local_v0.05
+test_batch_size: 128
+tied_noise: true
+train_batch_size: 64
+train_mode: true
+