added rvt.
Browse files- rvt/exp_cfg.yaml +33 -0
- rvt/model_14.pth +3 -0
- rvt/mvt_cfg.yaml +24 -0
rvt/exp_cfg.yaml
ADDED
|
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
agent: our
|
| 2 |
+
bs: 3
|
| 3 |
+
epochs: 15
|
| 4 |
+
exp_id: all
|
| 5 |
+
num_workers: 3
|
| 6 |
+
peract:
|
| 7 |
+
add_rgc_loss: true
|
| 8 |
+
lambda_weight_l2: 1.0e-06
|
| 9 |
+
lr: 0.0001
|
| 10 |
+
lr_cos_dec: true
|
| 11 |
+
num_rotation_classes: 72
|
| 12 |
+
optimizer_type: lamb
|
| 13 |
+
transform_augmentation: true
|
| 14 |
+
transform_augmentation_rpy:
|
| 15 |
+
- 0.0
|
| 16 |
+
- 0.0
|
| 17 |
+
- 45.0
|
| 18 |
+
transform_augmentation_xyz:
|
| 19 |
+
- 0.125
|
| 20 |
+
- 0.125
|
| 21 |
+
- 0.125
|
| 22 |
+
warmup_steps: 2000
|
| 23 |
+
rvt:
|
| 24 |
+
gt_hm_sigma: 1.5
|
| 25 |
+
img_aug: 0.1
|
| 26 |
+
move_pc_in_bound: true
|
| 27 |
+
place_with_mean: false
|
| 28 |
+
peract_official:
|
| 29 |
+
cfg_path: configs/peract_official_config.yaml
|
| 30 |
+
resume: ''
|
| 31 |
+
sample_distribution_mode: task_uniform
|
| 32 |
+
tasks: all
|
| 33 |
+
|
rvt/model_14.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d5192127574dd4e96a3a382fcf6112af2a185d9470b2522a4ac222c71ddd22e6
|
| 3 |
+
size 436405301
|
rvt/mvt_cfg.yaml
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
activation: lrelu
|
| 2 |
+
add_corr: true
|
| 3 |
+
add_depth: true
|
| 4 |
+
add_lang: true
|
| 5 |
+
add_pixel_loc: true
|
| 6 |
+
add_proprio: true
|
| 7 |
+
attn_dim: 512
|
| 8 |
+
attn_dim_head: 64
|
| 9 |
+
attn_dropout: 0.1
|
| 10 |
+
attn_heads: 8
|
| 11 |
+
decoder_dropout: 0.0
|
| 12 |
+
depth: 8
|
| 13 |
+
feat_dim: 220
|
| 14 |
+
final_dim: 64
|
| 15 |
+
im_channels: 64
|
| 16 |
+
img_feat_dim: 3
|
| 17 |
+
img_patch_size: 11
|
| 18 |
+
img_size: 220
|
| 19 |
+
lang_dim: 512
|
| 20 |
+
lang_len: 77
|
| 21 |
+
pe_fix: true
|
| 22 |
+
proprio_dim: 4
|
| 23 |
+
self_cross_ver: 1
|
| 24 |
+
weight_tie_layers: false
|