Skip to content

Commit

Permalink
conditional stuff seems to work
Browse files Browse the repository at this point in the history
  • Loading branch information
beneisner committed May 13, 2024
1 parent 45857ab commit 5bef65b
Show file tree
Hide file tree
Showing 63 changed files with 697 additions and 119 deletions.
8 changes: 8 additions & 0 deletions configs/commands/rlbench/_default/train_taxpose_tc.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_train@_here_
- override /model: taxpose_tc
- override /task: DEFAULT
- override /phase: all
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: pick_and_lift
- override /phase: final
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: pick_and_lift
- override /phase: grasp
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: pick_and_lift
- override /phase: lift
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
#!/bin/bash

set -e

echo 'Running precision eval'

echo "--------------------------------------------------------------------------------"
echo 'Evaluating pregrasp'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/pick_and_lift/taxpose_tc/precision_eval/pregrasp data_root=/data wandb.group=rlbench_pick_and_lift $@"

python scripts/eval_metrics.py --config-name commands/rlbench/pick_and_lift/taxpose_tc/precision_eval/pregrasp data_root=/data wandb.group=rlbench_pick_and_lift $@

echo "--------------------------------------------------------------------------------"
echo 'Evaluating grasp'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/pick_and_lift/taxpose_tc/precision_eval/grasp data_root=/data wandb.group=rlbench_pick_and_lift $@"

python scripts/eval_metrics.py --config-name commands/rlbench/pick_and_lift/taxpose_tc/precision_eval/grasp data_root=/data wandb.group=rlbench_pick_and_lift $@

echo "--------------------------------------------------------------------------------"
echo 'Evaluating lift'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/pick_and_lift/taxpose_tc/precision_eval/lift data_root=/data wandb.group=rlbench_pick_and_lift $@"

python scripts/eval_metrics.py --config-name commands/rlbench/pick_and_lift/taxpose_tc/precision_eval/lift data_root=/data wandb.group=rlbench_pick_and_lift $@

echo "--------------------------------------------------------------------------------"
echo 'Evaluating final'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/pick_and_lift/taxpose_tc/precision_eval/final data_root=/data wandb.group=rlbench_pick_and_lift $@"

python scripts/eval_metrics.py --config-name commands/rlbench/pick_and_lift/taxpose_tc/precision_eval/final data_root=/data wandb.group=rlbench_pick_and_lift $@
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: pick_and_lift
- override /phase: pregrasp
- _self_
8 changes: 8 additions & 0 deletions configs/commands/rlbench/pick_and_lift/train_taxpose_tc.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_train@_here_
- override /model: taxpose_tc
- override /task: pick_and_lift
- override /phase: all
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: pick_up_cup
- override /phase: grasp
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: pick_up_cup
- override /phase: lift
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
#!/bin/bash

set -e

echo 'Running precision eval'

echo "--------------------------------------------------------------------------------"
echo 'Evaluating pregrasp'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/pick_up_cup/taxpose_tc/precision_eval/pregrasp data_root=/data wandb.group=rlbench_pick_up_cup $@"

python scripts/eval_metrics.py --config-name commands/rlbench/pick_up_cup/taxpose_tc/precision_eval/pregrasp data_root=/data wandb.group=rlbench_pick_up_cup $@

echo "--------------------------------------------------------------------------------"
echo 'Evaluating grasp'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/pick_up_cup/taxpose_tc/precision_eval/grasp data_root=/data wandb.group=rlbench_pick_up_cup $@"

python scripts/eval_metrics.py --config-name commands/rlbench/pick_up_cup/taxpose_tc/precision_eval/grasp data_root=/data wandb.group=rlbench_pick_up_cup $@

echo "--------------------------------------------------------------------------------"
echo 'Evaluating lift'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/pick_up_cup/taxpose_tc/precision_eval/lift data_root=/data wandb.group=rlbench_pick_up_cup $@"

python scripts/eval_metrics.py --config-name commands/rlbench/pick_up_cup/taxpose_tc/precision_eval/lift data_root=/data wandb.group=rlbench_pick_up_cup $@
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: pick_up_cup
- override /phase: pregrasp
- _self_
8 changes: 8 additions & 0 deletions configs/commands/rlbench/pick_up_cup/train_taxpose_tc.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_train@_here_
- override /model: taxpose_tc
- override /task: pick_up_cup
- override /phase: all
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: push_button
- override /phase: postpush
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
#!/bin/bash

set -e

echo 'Running precision eval'

echo "--------------------------------------------------------------------------------"
echo 'Evaluating prepush'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/push_button/taxpose_tc/precision_eval/prepush data_root=/data wandb.group=rlbench_push_button $@"

python scripts/eval_metrics.py --config-name commands/rlbench/push_button/taxpose_tc/precision_eval/prepush data_root=/data wandb.group=rlbench_push_button $@

echo "--------------------------------------------------------------------------------"
echo 'Evaluating postpush'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/push_button/taxpose_tc/precision_eval/postpush data_root=/data wandb.group=rlbench_push_button $@"

python scripts/eval_metrics.py --config-name commands/rlbench/push_button/taxpose_tc/precision_eval/postpush data_root=/data wandb.group=rlbench_push_button $@
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: push_button
- override /phase: prepush
- _self_
8 changes: 8 additions & 0 deletions configs/commands/rlbench/push_button/train_taxpose_tc.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_train@_here_
- override /model: taxpose_tc
- override /task: push_button
- override /phase: all
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: put_knife_on_chopping_board
- override /phase: grasp
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: put_knife_on_chopping_board
- override /phase: lift
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: put_knife_on_chopping_board
- override /phase: place
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
#!/bin/bash

set -e

echo 'Running precision eval'

echo "--------------------------------------------------------------------------------"
echo 'Evaluating pregrasp'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/put_knife_on_chopping_board/taxpose_tc/precision_eval/pregrasp data_root=/data wandb.group=rlbench_put_knife_on_chopping_board $@"

python scripts/eval_metrics.py --config-name commands/rlbench/put_knife_on_chopping_board/taxpose_tc/precision_eval/pregrasp data_root=/data wandb.group=rlbench_put_knife_on_chopping_board $@

echo "--------------------------------------------------------------------------------"
echo 'Evaluating grasp'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/put_knife_on_chopping_board/taxpose_tc/precision_eval/grasp data_root=/data wandb.group=rlbench_put_knife_on_chopping_board $@"

python scripts/eval_metrics.py --config-name commands/rlbench/put_knife_on_chopping_board/taxpose_tc/precision_eval/grasp data_root=/data wandb.group=rlbench_put_knife_on_chopping_board $@

echo "--------------------------------------------------------------------------------"
echo 'Evaluating lift'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/put_knife_on_chopping_board/taxpose_tc/precision_eval/lift data_root=/data wandb.group=rlbench_put_knife_on_chopping_board $@"

python scripts/eval_metrics.py --config-name commands/rlbench/put_knife_on_chopping_board/taxpose_tc/precision_eval/lift data_root=/data wandb.group=rlbench_put_knife_on_chopping_board $@

echo "--------------------------------------------------------------------------------"
echo 'Evaluating place'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/put_knife_on_chopping_board/taxpose_tc/precision_eval/place data_root=/data wandb.group=rlbench_put_knife_on_chopping_board $@"

python scripts/eval_metrics.py --config-name commands/rlbench/put_knife_on_chopping_board/taxpose_tc/precision_eval/place data_root=/data wandb.group=rlbench_put_knife_on_chopping_board $@
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: put_knife_on_chopping_board
- override /phase: pregrasp
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_train@_here_
- override /model: taxpose_tc
- override /task: put_knife_on_chopping_board
- override /phase: all
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: put_money_in_safe
- override /phase: grasp
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: put_money_in_safe
- override /phase: lift
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: put_money_in_safe
- override /phase: place
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
#!/bin/bash

set -e

echo 'Running precision eval'

echo "--------------------------------------------------------------------------------"
echo 'Evaluating pregrasp'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/put_money_in_safe/taxpose_tc/precision_eval/pregrasp data_root=/data wandb.group=rlbench_put_money_in_safe $@"

python scripts/eval_metrics.py --config-name commands/rlbench/put_money_in_safe/taxpose_tc/precision_eval/pregrasp data_root=/data wandb.group=rlbench_put_money_in_safe $@

echo "--------------------------------------------------------------------------------"
echo 'Evaluating grasp'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/put_money_in_safe/taxpose_tc/precision_eval/grasp data_root=/data wandb.group=rlbench_put_money_in_safe $@"

python scripts/eval_metrics.py --config-name commands/rlbench/put_money_in_safe/taxpose_tc/precision_eval/grasp data_root=/data wandb.group=rlbench_put_money_in_safe $@

echo "--------------------------------------------------------------------------------"
echo 'Evaluating lift'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/put_money_in_safe/taxpose_tc/precision_eval/lift data_root=/data wandb.group=rlbench_put_money_in_safe $@"

python scripts/eval_metrics.py --config-name commands/rlbench/put_money_in_safe/taxpose_tc/precision_eval/lift data_root=/data wandb.group=rlbench_put_money_in_safe $@

echo "--------------------------------------------------------------------------------"
echo 'Evaluating preplace'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/put_money_in_safe/taxpose_tc/precision_eval/preplace data_root=/data wandb.group=rlbench_put_money_in_safe $@"

python scripts/eval_metrics.py --config-name commands/rlbench/put_money_in_safe/taxpose_tc/precision_eval/preplace data_root=/data wandb.group=rlbench_put_money_in_safe $@

echo "--------------------------------------------------------------------------------"
echo 'Evaluating place'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/put_money_in_safe/taxpose_tc/precision_eval/place data_root=/data wandb.group=rlbench_put_money_in_safe $@"

python scripts/eval_metrics.py --config-name commands/rlbench/put_money_in_safe/taxpose_tc/precision_eval/place data_root=/data wandb.group=rlbench_put_money_in_safe $@
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: put_money_in_safe
- override /phase: pregrasp
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: put_money_in_safe
- override /phase: preplace
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_train@_here_
- override /model: taxpose_tc
- override /task: put_money_in_safe
- override /phase: all
- _self_
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
#!/bin/bash

set -e

echo 'Running precision eval'

echo "--------------------------------------------------------------------------------"
echo 'Evaluating reach'
echo "--------------------------------------------------------------------------------"
echo "python scripts/eval_metrics.py --config-name commands/rlbench/reach_target/taxpose_tc/precision_eval/reach data_root=/data wandb.group=rlbench_reach_target $@"

python scripts/eval_metrics.py --config-name commands/rlbench/reach_target/taxpose_tc/precision_eval/reach data_root=/data wandb.group=rlbench_reach_target $@
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_

defaults:
- /commands/rlbench/_eval_metrics@_here_
- override /model: taxpose_tc
- override /task: reach_target
- override /phase: reach
- _self_
Loading

0 comments on commit 5bef65b

Please sign in to comment.