angoyal commited on
Commit
a01b10b
·
1 Parent(s): b11bedc

added rvt.

Browse files
Files changed (3) hide show
  1. rvt/exp_cfg.yaml +33 -0
  2. rvt/model_14.pth +3 -0
  3. 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