File size: 309 Bytes
75da528 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | 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"
|