Add Ours checkpoints (jta_ours, jrdb_ours, num_modes=1)

checkpoints/jrdb_ours/checkpoint.pth.tar

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

checkpoints/jrdb_ours/config.yaml

@@ -0,0 +1,35 @@
+DATA:
+  preprocessed: true
+  train_datasets:
+  - jrdb_all_visual_cues
+MODEL:
+  checkpoint: ''
+  dim_feedforward: 1024
+  dim_hidden: 128
+  eval_single: false
+  num_heads: 4
+  num_layers_global: 3
+  num_layers_local: 6
+  output_scale: 1
+  seq_len: 246
+  token_num: 26
+  type: transmotion
+  valuenet_checkpoint: /home/halo/plausibl/pacer/output/exp/pacer/valuenet_realpath_JTA+JRDB_valuenet_00025000.pth
+OUTPUT:
+  ckpt_dir: /misc/dl00/halo/plausibl/social-transmotion/experiments/JRDB/jrdb_value150/checkpoints
+  log_dir: /misc/dl00/halo/plausibl/social-transmotion/experiments/JRDB/jrdb_value150
+  runs_dir: /misc/dl00/halo/plausibl/social-transmotion/experiments/JRDB/jrdb_value150/tensorboard
+SEED: 0
+TRAIN:
+  aux_weight: 0.2
+  batch_size: 28
+  epochs: 150
+  input_track_size: 9
+  lr: 0.0001
+  lr_decay: 1
+  lr_drop: true
+  max_grad_norm: 1.0
+  num_workers: 3
+  optimizer: adam
+  output_track_size: 12
+  val_frequency: 2

checkpoints/jta_ours/checkpoint.pth.tar

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

checkpoints/jta_ours/config.yaml

@@ -0,0 +1,35 @@
+DATA:
+  preprocessed: true
+  train_datasets:
+  - jta_all_visual_cues
+MODEL:
+  checkpoint: /home/halo/plausibl/social-transmotion/experiments/jta_smpl/checkpoints/best_val_checkpoint.pth.tar
+  dim_feedforward: 1024
+  dim_hidden: 128
+  eval_single: false
+  num_heads: 4
+  num_layers_global: 3
+  num_layers_local: 6
+  output_scale: 1
+  seq_len: 453
+  token_num: 49
+  type: transmotion
+  valuenet_checkpoint: /home/halo/plausibl/pacer/output/exp/pacer/valuenet_realpath_JTA+JRDB_valuenet_00025000.pth
+OUTPUT:
+  ckpt_dir: /misc/dl00/halo/plausibl/social-transmotion/experiments/JTA/jta_valuenet_100/checkpoints
+  log_dir: /misc/dl00/halo/plausibl/social-transmotion/experiments/JTA/jta_valuenet_100
+  runs_dir: /misc/dl00/halo/plausibl/social-transmotion/experiments/JTA/jta_valuenet_100/tensorboard
+SEED: 0
+TRAIN:
+  aux_weight: 0.2
+  batch_size: 13
+  epochs: 50
+  input_track_size: 9
+  lr: 0.0001
+  lr_decay: 1
+  lr_drop: true
+  max_grad_norm: 1.0
+  num_workers: 3
+  optimizer: adam
+  output_track_size: 12
+  val_frequency: 2