sam2act_robomme / sam2act_plus_all_v4 /exp_cfg_plus.yaml
HongzeFu's picture
Add sam2act_plus_all_v4 (SAM2Act+, full 10 epochs)
61cd3ae verified
Raw
History Blame Contribute Delete
906 Bytes
agent: our
bs: 20
demo: 100
epochs: 10
exp_id: sam2act
exp_name: plus_all_v4
num_workers: 5
peract:
add_rgc_loss: true
amp: true
bnb: true
lambda_weight_l2: 0.0001
lr: 1.25e-06
lr_cos_dec: true
num_rotation_classes: 72
optimizer_type: lamb
same_trans_aug_per_seq: false
transform_augmentation: true
transform_augmentation_rpy:
- 0.0
- 0.0
- 45.0
transform_augmentation_xyz:
- 0.125
- 0.125
- 0.125
warmup_steps: 2000
peract_official:
cfg_path: configs/peract_official_config.yaml
resume: ''
rvt:
gt_hm_sigma: 1.5
img_aug: 0.0
move_pc_in_bound: true
place_with_mean: false
sample_distribution_mode: demo_uniform_temporal
tasks: PickXtimes,StopCube,SwingXtimes,BinFill,VideoUnmaskSwap,VideoUnmask,ButtonUnmaskSwap,ButtonUnmask,VideoRepick,VideoPlaceButton,VideoPlaceOrder,PickHighlight,InsertPeg,MoveCube,PatternLock,RouteStick
train_iter: 800000
wandb: true