English
human motion
collision
PoseShield / config.yaml
ZYYY99's picture
Upload 4 files
75da528 verified
Raw
History Blame Contribute Delete
309 Bytes
TRAIN:
NUM_EPOCHS: 300
BATCH_SIZE: 256
LR: 0.0002
VAL_INTERVAL: 10
EXP_NAME: "train_active_large_v2"
GRAD_LOSS_WEIGHT: 0.1
TD_LOSS_WEIGHT: 1.0
DT: 0.01
MODEL:
TYPE: "ResidualMLP"
IN_DIM: 126 # 21 joints × 6 values each
HIDDEN_DIM: 512
NUM_LAYERS: 12
ACTIVATION: "relu"