Skip to content

Commit

Permalink
tiago configs - FMR and FM
Browse files Browse the repository at this point in the history
  • Loading branch information
michalvavrecka committed Feb 1, 2024
1 parent efa6a7e commit 69e74db
Showing 1 changed file with 55 additions and 0 deletions.
55 changes: 55 additions & 0 deletions myGym/configs/train_FM_tiago.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,55 @@
{
#Environment
"env_name" :"Gym-v0",
"workspace" :"table_tiago",
"engine" :"pybullet",
"render" :"opengl",
"seed" :1,
"camera" :0,
"gui" :0,
"visualize" :0,
"visgym" :0,
#Robot
"robot" :"tiago_dual",
"robot_action" :"joints",
"robot_init" :[0.17, 0.33, 0.4],
"max_velocity" :5,
"max_force" :300,
"action_repeat" :1,
#Task
"task_type" :"FM",
"natural_language" :0,
"task_objects" :[{"init":{"obj_name":"sponge","fixed":0,"rand_rot":0, "sampling_area":[0.2,0.4,0.4,0.6,-0.3,-0.3]},
"goal":{"obj_name":"cube_holes","fixed":1,"rand_rot":0, "sampling_area":[-0.7, -0.75, 0.2, -0.5, -0.3, 0.4]}}],
"color_dict" : {"cube_holes":["green"], "target":["gray"]},
"used_objects" :{"num_range":[0,0], "obj_list":[]},
// Observation
// actual_state options (pick one): "endeff_xyz", "endeff_6D" (robot end effector), "obj_xyz", "obj_6D", "vae", "yolact", "voxel" or "dope"
// goal_state options (pick one): "obj_xyz", "obj_6D", "vae", "yolact", "voxel" or "dope"
// additional_obs options (pick none/one/more): "joints_xyz", "joints_angles", "endeff_xyz", "endeff_6D", "touch", "distractor"
"observation" : {"actual_state":"obj_6D", "goal_state":"obj_6D", "additional_obs":["endeff_xyz"]},
#Distractor
"distractors" : {"list":null, "moveable":1, "constant_speed":0, "movement_dims":3,
"movement_endpoints":[-0.3, 0.3, 0.4, 0.7, 0.1, 0.3]},
#Reward
"reward" :"FM",
"distance_type" :"euclidean",
"vae_path" :null,
"yolact_path" :null,
"yolact_config" :null,
#Train
"train_framework" :"tensorflow",
"algo" :"multippo2",
"num_networks" :2,
"max_episode_steps" :512000,
"algo_steps" :512,
"steps" :5000000,
"pretrained_model" :null,
"multiprocessing" :false,
#Evaluation
"eval_freq" :100000,
"eval_episodes" :30,
#Saving and logging
"logdir" :"trained_models/swipe",
"record" :0
}

0 comments on commit 69e74db

Please sign in to comment.