Skip to content

Commit

Permalink
Update example scripts
Browse files Browse the repository at this point in the history
Signed-off-by: Andrej Orsula <[email protected]>
  • Loading branch information
AndrejOrsula committed Dec 16, 2022
1 parent b95ae43 commit 14f5402
Show file tree
Hide file tree
Showing 6 changed files with 26 additions and 14 deletions.
8 changes: 5 additions & 3 deletions examples/ex_evaluate.bash
Original file line number Diff line number Diff line change
Expand Up @@ -23,10 +23,12 @@ ROBOT_MODEL="panda"
# ENV="Reach-OctreeWithIntensity-Gazebo-v0"
# ENV="Reach-OctreeWithColor-Gazebo-v0"
## Grasp
ENV="Grasp-Gazebo-v0"
# ENV="Grasp-Octree-Gazebo-v0"
# ENV="Grasp-OctreeWithIntensity-Gazebo-v0"
ENV="Grasp-OctreeWithColor-Gazebo-v0"
# ENV="Grasp-OctreeWithColor-Gazebo-v0"
## GraspPlanetary
# ENV="GraspPlanetary-Gazebo-v0"
# ENV="GraspPlanetary-MonoImage-Gazebo-v0"
# ENV="GraspPlanetary-ColorImage-Gazebo-v0"
# ENV="GraspPlanetary-DepthImage-Gazebo-v0"
Expand All @@ -37,9 +39,9 @@ ENV="Grasp-OctreeWithColor-Gazebo-v0"
# ENV="GraspPlanetary-OctreeWithColor-Gazebo-v0"

## Selection of RL algorithm
ALGO="sac"
# ALGO="td3"
# ALGO="sac"
ALGO="tqc"
# ALGO="tqc"

## Path to logs directory
LOG_FOLDER="${PWD}/drl_grasping_training/train/${ENV}/logs"
Expand Down
8 changes: 5 additions & 3 deletions examples/ex_evaluate_pretrained_agent.bash
Original file line number Diff line number Diff line change
Expand Up @@ -22,10 +22,12 @@ ROBOT_MODEL="panda"
# ENV="Reach-OctreeWithIntensity-Gazebo-v0"
# ENV="Reach-OctreeWithColor-Gazebo-v0"
## Grasp
ENV="Grasp-Gazebo-v0"
# ENV="Grasp-Octree-Gazebo-v0"
# ENV="Grasp-OctreeWithIntensity-Gazebo-v0"
ENV="Grasp-OctreeWithColor-Gazebo-v0"
# ENV="Grasp-OctreeWithColor-Gazebo-v0"
## GraspPlanetary
# ENV="GraspPlanetary-Gazebo-v0"
# ENV="GraspPlanetary-MonoImage-Gazebo-v0"
# ENV="GraspPlanetary-ColorImage-Gazebo-v0"
# ENV="GraspPlanetary-DepthImage-Gazebo-v0"
Expand All @@ -36,9 +38,9 @@ ENV="Grasp-OctreeWithColor-Gazebo-v0"
# ENV="GraspPlanetary-OctreeWithColor-Gazebo-v0"

## Selection of RL algorithm
ALGO="sac"
# ALGO="td3"
# ALGO="sac"
ALGO="tqc"
# ALGO="tqc"

## Path to logs directory
LOG_FOLDER="$(ros2 pkg prefix --share drl_grasping)/pretrained_agents"
Expand Down
6 changes: 4 additions & 2 deletions examples/ex_evaluate_real.bash
Original file line number Diff line number Diff line change
Expand Up @@ -23,10 +23,12 @@ ROBOT_MODEL="panda"
# ENV="Reach-OctreeWithIntensity-v0"
# ENV="Reach-OctreeWithColor-v0"
## Grasp
# ENV="Grasp-v0"
# ENV="Grasp-Octree-v0"
# ENV="Grasp-OctreeWithIntensity-v0"
ENV="Grasp-OctreeWithColor-v0"
## GraspPlanetary
# ENV="GraspPlanetary-v0"
# ENV="GraspPlanetary-MonoImage-v0"
# ENV="GraspPlanetary-ColorImage-v0"
# ENV="GraspPlanetary-DepthImage-v0"
Expand All @@ -37,9 +39,9 @@ ENV="Grasp-OctreeWithColor-v0"
# ENV="GraspPlanetary-OctreeWithColor-v0"

## Selection of RL algorithm
ALGO="sac"
# ALGO="td3"
# ALGO="sac"
ALGO="tqc"
# ALGO="tqc"

## Path to logs directory
LOG_FOLDER="${PWD}/drl_grasping_training/train/${ENV}/logs"
Expand Down
8 changes: 5 additions & 3 deletions examples/ex_optimize.bash
Original file line number Diff line number Diff line change
Expand Up @@ -23,10 +23,12 @@ ROBOT_MODEL="panda"
# ENV="Reach-OctreeWithIntensity-Gazebo-v0"
# ENV="Reach-OctreeWithColor-Gazebo-v0"
## Grasp
ENV="Grasp-Gazebo-v0"
# ENV="Grasp-Octree-Gazebo-v0"
# ENV="Grasp-OctreeWithIntensity-Gazebo-v0"
ENV="Grasp-OctreeWithColor-Gazebo-v0"
# ENV="Grasp-OctreeWithColor-Gazebo-v0"
## GraspPlanetary
# ENV="GraspPlanetary-Gazebo-v0"
# ENV="GraspPlanetary-MonoImage-Gazebo-v0"
# ENV="GraspPlanetary-ColorImage-Gazebo-v0"
# ENV="GraspPlanetary-DepthImage-Gazebo-v0"
Expand All @@ -37,9 +39,9 @@ ENV="Grasp-OctreeWithColor-Gazebo-v0"
# ENV="GraspPlanetary-OctreeWithColor-Gazebo-v0"

## Selection of RL algorithm
ALGO="sac"
# ALGO="td3"
# ALGO="sac"
ALGO="tqc"
# ALGO="tqc"

## Path to logs directory
LOG_FOLDER="${PWD}/drl_grasping_training/optimize/${ENV}/logs"
Expand Down
2 changes: 2 additions & 0 deletions examples/ex_random_agent.bash
Original file line number Diff line number Diff line change
Expand Up @@ -20,10 +20,12 @@ ROBOT_MODEL="lunalab_summit_xl_gen"
# ENV="Reach-OctreeWithIntensity-Gazebo-v0"
# ENV="Reach-OctreeWithColor-Gazebo-v0"
## Grasp
# ENV="Grasp-Gazebo-v0"
# ENV="Grasp-Octree-Gazebo-v0"
# ENV="Grasp-OctreeWithIntensity-Gazebo-v0"
# ENV="Grasp-OctreeWithColor-Gazebo-v0"
## GraspPlanetary
# ENV="GraspPlanetary-Gazebo-v0"
# ENV="GraspPlanetary-MonoImage-Gazebo-v0"
# ENV="GraspPlanetary-ColorImage-Gazebo-v0"
# ENV="GraspPlanetary-DepthImage-Gazebo-v0"
Expand Down
8 changes: 5 additions & 3 deletions examples/ex_train.bash
Original file line number Diff line number Diff line change
Expand Up @@ -23,10 +23,12 @@ ROBOT_MODEL="panda"
# ENV="Reach-OctreeWithIntensity-Gazebo-v0"
# ENV="Reach-OctreeWithColor-Gazebo-v0"
## Grasp
ENV="Grasp-Gazebo-v0"
# ENV="Grasp-Octree-Gazebo-v0"
# ENV="Grasp-OctreeWithIntensity-Gazebo-v0"
ENV="Grasp-OctreeWithColor-Gazebo-v0"
# ENV="Grasp-OctreeWithColor-Gazebo-v0"
## GraspPlanetary
# ENV="GraspPlanetary-Gazebo-v0"
# ENV="GraspPlanetary-MonoImage-Gazebo-v0"
# ENV="GraspPlanetary-ColorImage-Gazebo-v0"
# ENV="GraspPlanetary-DepthImage-Gazebo-v0"
Expand All @@ -37,9 +39,9 @@ ENV="Grasp-OctreeWithColor-Gazebo-v0"
# ENV="GraspPlanetary-OctreeWithColor-Gazebo-v0"

## Selection of RL algorithm
ALGO="sac"
# ALGO="td3"
# ALGO="sac"
ALGO="tqc"
# ALGO="tqc"

## Path to logs directory
LOG_FOLDER="${PWD}/drl_grasping_training/train/${ENV}/logs"
Expand Down

0 comments on commit 14f5402

Please sign in to comment.