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