| 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 | |