Upload folder using huggingface_hub
Browse files- args.yaml +8 -0
- eval/test_4/1/eval_config.yaml +20 -0
- eval/test_4/1/events.out.tfevents.1711381746.df8adc5b96ee.784750.0 +3 -0
- eval/test_4/1/model_4/eval_results.csv +19 -0
- eval/test_4/2/eval_config.yaml +20 -0
- eval/test_4/2/events.out.tfevents.1711416669.df8adc5b96ee.786462.0 +3 -0
- eval/test_4/2/model_4/eval_results.csv +19 -0
- eval/test_4/3/eval_config.yaml +20 -0
- eval/test_4/3/events.out.tfevents.1711427679.df8adc5b96ee.787370.0 +3 -0
- eval/test_4/3/model_4/eval_results.csv +19 -0
- eval/test_4/model_4_final_scores.txt +10 -0
- exp_cfg.yaml +41 -0
- model_4.pth +3 -0
- mvt_cfg.yaml +29 -0
args.yaml
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
device: 0,1,2,3,4,5,6,7
|
| 2 |
+
exp_cfg_opts: ''
|
| 3 |
+
exp_cfg_path: configs/all.yaml
|
| 4 |
+
log_dir: runs/test
|
| 5 |
+
mvt_cfg_opts: ''
|
| 6 |
+
mvt_cfg_path: ''
|
| 7 |
+
refresh_replay: false
|
| 8 |
+
with_eval: false
|
eval/test_4/1/eval_config.yaml
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
device: 2
|
| 2 |
+
episode_length: 25
|
| 3 |
+
eval_datafolder: /remote-home/share/RLBench/test
|
| 4 |
+
eval_episodes: 25
|
| 5 |
+
eval_log_dir: runs/keyframes_bs.3.8.lr.5e-5_r3m_v3.depth.1.dim.512/eval/test_4/1
|
| 6 |
+
exp_cfg_path: null
|
| 7 |
+
ground_truth: false
|
| 8 |
+
headless: true
|
| 9 |
+
log_name: test_4/1
|
| 10 |
+
model_folder: runs/keyframes_bs.3.8.lr.5e-5_r3m_v3.depth.1.dim.512
|
| 11 |
+
model_name: model_4.pth
|
| 12 |
+
mvt_cfg_path: null
|
| 13 |
+
peract_model_dir: runs/peract_official/seed0/weights/600000
|
| 14 |
+
peract_official: false
|
| 15 |
+
save_video: false
|
| 16 |
+
skip: false
|
| 17 |
+
start_episode: 0
|
| 18 |
+
tasks:
|
| 19 |
+
- all
|
| 20 |
+
use_input_place_with_mean: false
|
eval/test_4/1/events.out.tfevents.1711381746.df8adc5b96ee.784750.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bc6b03bdccea2f148eb2c602e37435c8471ac42e8653c3369648b50a74b73631
|
| 3 |
+
size 134
|
eval/test_4/1/model_4/eval_results.csv
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
task,success rate,length,total_transitions
|
| 2 |
+
put_item_in_drawer,12.0,6.72,168
|
| 3 |
+
reach_and_drag,92.0,7.04,344
|
| 4 |
+
turn_tap,72.0,7.88,541
|
| 5 |
+
slide_block_to_color_target,56.0,11.44,827
|
| 6 |
+
open_drawer,80.0,5.0,952
|
| 7 |
+
put_groceries_in_cupboard,24.0,16.08,1354
|
| 8 |
+
place_shape_in_shape_sorter,24.0,21.04,1880
|
| 9 |
+
put_money_in_safe,88.0,8.2,2085
|
| 10 |
+
push_buttons,100.0,3.68,2177
|
| 11 |
+
close_jar,76.0,10.56,2441
|
| 12 |
+
stack_blocks,20.0,19.84,2937
|
| 13 |
+
place_cups,0.0,24.64,3553
|
| 14 |
+
place_wine_at_rack_location,80.0,9.0,3778
|
| 15 |
+
light_bulb_in,24.0,20.56,4292
|
| 16 |
+
sweep_to_dustpan_of_size,32.0,18.6,4757
|
| 17 |
+
insert_onto_square_peg,16.0,21.76,5301
|
| 18 |
+
meat_off_grill,96.0,5.8,5446
|
| 19 |
+
stack_cups,0.0,22.92,6019
|
eval/test_4/2/eval_config.yaml
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
device: 2
|
| 2 |
+
episode_length: 25
|
| 3 |
+
eval_datafolder: /remote-home/share/RLBench/test
|
| 4 |
+
eval_episodes: 25
|
| 5 |
+
eval_log_dir: runs/keyframes_bs.3.8.lr.5e-5_r3m_v3.depth.1.dim.512/eval/test_4/2
|
| 6 |
+
exp_cfg_path: null
|
| 7 |
+
ground_truth: false
|
| 8 |
+
headless: true
|
| 9 |
+
log_name: test_4/2
|
| 10 |
+
model_folder: runs/keyframes_bs.3.8.lr.5e-5_r3m_v3.depth.1.dim.512
|
| 11 |
+
model_name: model_4.pth
|
| 12 |
+
mvt_cfg_path: null
|
| 13 |
+
peract_model_dir: runs/peract_official/seed0/weights/600000
|
| 14 |
+
peract_official: false
|
| 15 |
+
save_video: false
|
| 16 |
+
skip: false
|
| 17 |
+
start_episode: 0
|
| 18 |
+
tasks:
|
| 19 |
+
- all
|
| 20 |
+
use_input_place_with_mean: false
|
eval/test_4/2/events.out.tfevents.1711416669.df8adc5b96ee.786462.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:04b1719c2cd09241ab4a2975bd8db0e0c694a2875f81c568fbbdea82a232df96
|
| 3 |
+
size 134
|
eval/test_4/2/model_4/eval_results.csv
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
task,success rate,length,total_transitions
|
| 2 |
+
put_item_in_drawer,20.0,6.6,165
|
| 3 |
+
reach_and_drag,96.0,6.2,320
|
| 4 |
+
turn_tap,68.0,7.6,510
|
| 5 |
+
slide_block_to_color_target,64.0,10.52,773
|
| 6 |
+
open_drawer,84.0,5.44,909
|
| 7 |
+
put_groceries_in_cupboard,24.0,14.2,1264
|
| 8 |
+
place_shape_in_shape_sorter,20.0,19.8,1759
|
| 9 |
+
put_money_in_safe,92.0,6.88,1931
|
| 10 |
+
push_buttons,100.0,3.52,2019
|
| 11 |
+
close_jar,76.0,10.56,2283
|
| 12 |
+
stack_blocks,16.0,21.04,2809
|
| 13 |
+
place_cups,0.0,23.92,3407
|
| 14 |
+
place_wine_at_rack_location,80.0,9.0,3632
|
| 15 |
+
light_bulb_in,36.0,18.16,4086
|
| 16 |
+
sweep_to_dustpan_of_size,36.0,17.8,4531
|
| 17 |
+
insert_onto_square_peg,16.0,20.44,5042
|
| 18 |
+
meat_off_grill,96.0,5.8,5187
|
| 19 |
+
stack_cups,0.0,24.52,5800
|
eval/test_4/3/eval_config.yaml
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
device: 2
|
| 2 |
+
episode_length: 25
|
| 3 |
+
eval_datafolder: /remote-home/share/RLBench/test
|
| 4 |
+
eval_episodes: 25
|
| 5 |
+
eval_log_dir: runs/keyframes_bs.3.8.lr.5e-5_r3m_v3.depth.1.dim.512/eval/test_4/3
|
| 6 |
+
exp_cfg_path: null
|
| 7 |
+
ground_truth: false
|
| 8 |
+
headless: true
|
| 9 |
+
log_name: test_4/3
|
| 10 |
+
model_folder: runs/keyframes_bs.3.8.lr.5e-5_r3m_v3.depth.1.dim.512
|
| 11 |
+
model_name: model_4.pth
|
| 12 |
+
mvt_cfg_path: null
|
| 13 |
+
peract_model_dir: runs/peract_official/seed0/weights/600000
|
| 14 |
+
peract_official: false
|
| 15 |
+
save_video: false
|
| 16 |
+
skip: false
|
| 17 |
+
start_episode: 0
|
| 18 |
+
tasks:
|
| 19 |
+
- all
|
| 20 |
+
use_input_place_with_mean: false
|
eval/test_4/3/events.out.tfevents.1711427679.df8adc5b96ee.787370.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:eb4f7b4096833e483808b659dc23ed6ea39326083fd6bc7ab3f5da885c22d1ba
|
| 3 |
+
size 134
|
eval/test_4/3/model_4/eval_results.csv
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
task,success rate,length,total_transitions
|
| 2 |
+
put_item_in_drawer,12.0,4.44,111
|
| 3 |
+
reach_and_drag,100.0,6.12,264
|
| 4 |
+
turn_tap,76.0,6.2,419
|
| 5 |
+
slide_block_to_color_target,60.0,10.56,683
|
| 6 |
+
open_drawer,76.0,6.0,833
|
| 7 |
+
put_groceries_in_cupboard,24.0,15.84,1229
|
| 8 |
+
place_shape_in_shape_sorter,16.0,21.76,1773
|
| 9 |
+
put_money_in_safe,92.0,7.16,1952
|
| 10 |
+
push_buttons,100.0,3.68,2044
|
| 11 |
+
close_jar,72.0,11.32,2327
|
| 12 |
+
stack_blocks,12.0,19.48,2814
|
| 13 |
+
place_cups,0.0,24.6,3429
|
| 14 |
+
place_wine_at_rack_location,72.0,9.8,3674
|
| 15 |
+
light_bulb_in,32.0,19.12,4152
|
| 16 |
+
sweep_to_dustpan_of_size,40.0,17.04,4578
|
| 17 |
+
insert_onto_square_peg,20.0,21.12,5106
|
| 18 |
+
meat_off_grill,96.0,6.24,5262
|
| 19 |
+
stack_cups,0.0,23.92,5860
|
eval/test_4/model_4_final_scores.txt
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
Avg: 49.56; 12.0, 92.0, 72.0, 56.0, 80.0, 24.0, 24.0, 88.0, 100.0, 76.0, 20.0, 0.0, 80.0, 24.0, 32.0, 16.0, 96.0, 0.0
|
| 2 |
+
Avg: 51.33; 20.0, 96.0, 68.0, 64.0, 84.0, 24.0, 20.0, 92.0, 100.0, 76.0, 16.0, 0.0, 80.0, 36.0, 36.0, 16.0, 96.0, 0.0
|
| 3 |
+
Avg: 50.00; 12.0, 100.0, 76.0, 60.0, 76.0, 24.0, 16.0, 92.0, 100.0, 72.0, 12.0, 0.0, 72.0, 32.0, 40.0, 20.0, 96.0, 0.0
|
| 4 |
+
|
| 5 |
+
14.7 & 96.0 & 72.0 & 60.0 & 80.0 & 24.0 & 20.0 & 90.7 & 100.0 & 74.7 & 16.0 & 0.0 & 77.3 & 30.7 & 36.0 & 17.3 & 96.0 & 0.0 & 50.3
|
| 6 |
+
4.6 & 4.0 & 4.0 & 4.0 & 4.0 & 0.0 & 4.0 & 2.3 & 0.0 & 2.3 & 4.0 & 0.0 & 4.6 & 6.1 & 4.0 & 2.3 & 0.0 & 0.0 & 0.9
|
| 7 |
+
|
| 8 |
+
14.7_{\pm 4.6} & 96.0_{\pm 4.0} & 72.0_{\pm 4.0} & 60.0_{\pm 4.0} & 80.0_{\pm 4.0} & 24.0_{\pm 0.0} & 20.0_{\pm 4.0} & 90.7_{\pm 2.3} & 100.0_{\pm 0.0} & 74.7_{\pm 2.3} & 16.0_{\pm 4.0} & 0.0_{\pm 0.0} & 77.3_{\pm 4.6} & 30.7_{\pm 6.1} & 36.0_{\pm 4.0} & 17.3_{\pm 2.3} & 96.0_{\pm 0.0} & 0.0_{\pm 0.0} & 50.3 \pm 0.9
|
| 9 |
+
|
| 10 |
+
final score ::: 50.30
|
exp_cfg.yaml
ADDED
|
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
adapter: []
|
| 2 |
+
agent: our
|
| 3 |
+
attn_dim: 512
|
| 4 |
+
bs: 3
|
| 5 |
+
depth: 1
|
| 6 |
+
ds_rate: 1.0
|
| 7 |
+
epochs: 15
|
| 8 |
+
exp_id: rvt
|
| 9 |
+
model: MVT_Resnet
|
| 10 |
+
output_dim: 512
|
| 11 |
+
num_workers: 4
|
| 12 |
+
overwriter_log_dir: runs/UnadaptedR3M2RLBench
|
| 13 |
+
peract:
|
| 14 |
+
add_rgc_loss: true
|
| 15 |
+
lambda_weight_l2: 1.0e-06
|
| 16 |
+
lr: 5.0e-05
|
| 17 |
+
lr_cos_dec: true
|
| 18 |
+
num_rotation_classes: 72
|
| 19 |
+
optimizer_type: lamb
|
| 20 |
+
transform_augmentation: true
|
| 21 |
+
transform_augmentation_rpy:
|
| 22 |
+
- 0.0
|
| 23 |
+
- 0.0
|
| 24 |
+
- 45.0
|
| 25 |
+
transform_augmentation_xyz:
|
| 26 |
+
- 0.125
|
| 27 |
+
- 0.125
|
| 28 |
+
- 0.125
|
| 29 |
+
warmup_steps: 2000
|
| 30 |
+
peract_official:
|
| 31 |
+
cfg_path: configs/peract_official_config.yaml
|
| 32 |
+
pretrain: ../pretrains/UnadaptedR3M.pt
|
| 33 |
+
resume: ''
|
| 34 |
+
rvt:
|
| 35 |
+
gt_hm_sigma: 1.5
|
| 36 |
+
img_aug: 0.1
|
| 37 |
+
move_pc_in_bound: true
|
| 38 |
+
place_with_mean: false
|
| 39 |
+
sample_distribution_mode: task_uniform
|
| 40 |
+
tasks: all
|
| 41 |
+
|
model_4.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f2eb5c4fd2e46102b17258c888063f5ca5ff7d5874abcab2032387108f816951
|
| 3 |
+
size 636874353
|
mvt_cfg.yaml
ADDED
|
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
activation: lrelu
|
| 2 |
+
adapter: []
|
| 3 |
+
add_corr: true
|
| 4 |
+
add_depth: true
|
| 5 |
+
add_lang: true
|
| 6 |
+
add_pixel_loc: true
|
| 7 |
+
add_proprio: true
|
| 8 |
+
attn_dim: 512
|
| 9 |
+
attn_dim_head: 64
|
| 10 |
+
attn_dropout: 0.1
|
| 11 |
+
attn_heads: 8
|
| 12 |
+
decoder_dropout: 0.0
|
| 13 |
+
depth: 1
|
| 14 |
+
ds_rate: 1.0
|
| 15 |
+
feat_dim: 220
|
| 16 |
+
final_dim: 64
|
| 17 |
+
im_channels: 64
|
| 18 |
+
img_feat_dim: 3
|
| 19 |
+
img_patch_size: 11
|
| 20 |
+
img_size: 220
|
| 21 |
+
lang_dim: 512
|
| 22 |
+
lang_len: 77
|
| 23 |
+
model: MVT_Resnet
|
| 24 |
+
output_dim: 512
|
| 25 |
+
pe_fix: true
|
| 26 |
+
proprio_dim: 4
|
| 27 |
+
self_cross_ver: 1
|
| 28 |
+
weight_tie_layers: false
|
| 29 |
+
|