MCPcopy Create free account

hub / github.com/abalakrishna123/recovery-rl / functions

Functions314 in github.com/abalakrishna123/recovery-rl

⨍ Functions 314◇ Types & classes 47
Methodforward
(self, state, action)
recovery_rl/model.py:139
Methodforward
(self, state, action)
recovery_rl/model.py:188
Methodforward
(self, state, action)
recovery_rl/model.py:262
Methodforward
(self, prev_hidden, action)
recovery_rl/model.py:727
Methodforward
(self, observation)
recovery_rl/model.py:769
Methodforward
(self, hidden)
recovery_rl/model.py:821
Methodforward
(self, inputs, ret_logvar=False)
config/image_maze.py:71
Methodforward
(self, inputs, ret_logvar=False)
config/obj_dynamic_extraction.py:71
Methodforward
(self, inputs, ret_logvar=False)
config/obj_extraction.py:71
Methodforward
(self, inputs, ret_logvar=False)
config/navigation1.py:71
Methodforward
(self, inputs, ret_logvar=False)
config/navigation2.py:71
Methodget_desig_pix
(self, target_width, round=True, obj_poses=None)
env/base_mujoco_env.py:99
Methodget_encoding
Get encoded version of image
recovery_rl/VisualMPC.py:257
Methodget_goal_pix
(self, target_width, round=True)
env/base_mujoco_env.py:120
Functionget_offline_data
(num_transitions, images=False, save_rollouts=False,
env/image_maze.py:34
Functionget_offline_data
(num_transitions, task_demos=False, save_rollouts=False)
env/navigation1.py:133
Functionget_offline_data
(num_transitions, images=False, save_rollouts=False)
env/maze.py:34
Functionget_offline_data
(num_transitions, task_demos=False, save_rollouts=False)
env/navigation2.py:133
Methodgoal_reached
Checks if the environment hit a goal (if environment has goals) - e.x. if goal is to lift object should return true if object lif
env/base_env.py:50
Functionhard_update
(target, source)
recovery_rl/model.py:37
Methodhas_goal
:return: Whether or not environment has a goal
env/base_env.py:58
Methodis_stable
(self, s)
env/navigation1.py:129
Methodis_stable
(self, s)
env/navigation2.py:129
Methodjaw_width
(self)
env/obj_dynamic_extraction.py:185
Methodjaw_width
(self)
env/obj_extraction.py:146
Functionlogsumexp
(inputs, dim=None, keepdim=False)
recovery_rl/utils.py:30
Functionmake_bool
(arg)
config/default.py:122
Methodncam
(self)
env/base_mujoco_env.py:165
Methodnn_constructor
(self, model_init_cfg)
config/image_maze.py:151
Methodnn_constructor
(self, model_init_cfg)
config/obj_dynamic_extraction.py:151
Methodnn_constructor
(self, model_init_cfg)
config/obj_extraction.py:151
Methodnn_constructor
(self, model_init_cfg)
config/navigation1.py:155
Methodnn_constructor
(self, model_init_cfg)
config/maze.py:151
Methodnn_constructor
(self, model_init_cfg)
config/navigation2.py:155
Methodobject_poses
(self)
env/obj_dynamic_extraction.py:260
Methodobject_poses
(self)
env/obj_extraction.py:215
Methodobs_cost_fn
(self, obs)
config/image_maze.py:144
Methodobs_cost_fn
(self, obs)
config/obj_dynamic_extraction.py:144
Methodobs_cost_fn
(self, obs)
config/obj_extraction.py:144
Methodobs_cost_fn
(self, obs)
config/navigation1.py:142
Methodobs_cost_fn
(self, obs)
config/maze.py:144
Methodobs_cost_fn
(self, obs)
config/navigation2.py:142
Methodobs_postproc
(obs, pred)
config/image_maze.py:134
Methodobs_postproc
(obs, pred)
config/obj_dynamic_extraction.py:134
Methodobs_postproc
(obs, pred)
config/obj_extraction.py:134
Methodobs_postproc
(obs, pred)
config/navigation1.py:132
Methodobs_postproc
(obs, pred)
config/navigation2.py:132
Methodobtain_solution
Optimizes the cost function using the provided initial candidate distribution Arguments: init_mean (np.ndarray): The mean of the
recovery_rl/optimizers.py:73
Methodplot_trajectory
(self, states=None)
env/navigation1.py:121
Methodplot_trajectory
(self, states=None)
env/navigation2.py:121
Methodposition
(self)
env/obj_dynamic_extraction.py:256
Methodposition
(self)
env/obj_extraction.py:211
Functionprocess_obs
(obs)
recovery_rl/VisualMPC.py:65
Methodpush
(self, state, action, reward, next_state, done)
recovery_rl/replay_memory.py:47
Functionread_only
(arg)
config/default.py:130
Methodrecovery_thresh
(state, action)
recovery_rl/experiment.py:547
Functionrender
(loc)
env/navigation1.py:167
Functionrender
(loc)
env/navigation2.py:246
Methodrender
Renders the enviornment. Implements custom rendering support. If mode is: - dual: renders both left and main cameras - left:
env/base_mujoco_env.py:49
Methodreset
(self)
env/base_mujoco_env.py:46
Methodreset
Resets the environment. Returns initial observation as well as information needed to recreate initialization :return: obs dict (look
env/base_env.py:31
Methodreset
(self)
env/navigation1.py:91
Methodreset
(self)
env/navigation2.py:90
Methodreset
Resets this controller.
recovery_rl/MPC.py:34
Methodreset
(self)
recovery_rl/optimizers.py:21
Methodreset
(self)
recovery_rl/optimizers.py:70
Methodreset
Resets this controller.
recovery_rl/VisualMPC.py:39
Methodsample
samples a random action from the action space.
env/navigation2.py:115
Methodsample
(self, batch_size, pos_fraction=None)
recovery_rl/replay_memory.py:54
Methodsample
(self, state)
recovery_rl/model.py:425
Methodsample
(self, state)
recovery_rl/model.py:474
Methodsample
(self, state)
recovery_rl/model.py:522
Methodsample
(self, state)
recovery_rl/model.py:610
Methodsample
(self, state)
recovery_rl/model.py:699
Methodsdim
(self)
env/base_mujoco_env.py:161
Methodsdim
:return: Environment's state dimension
env/base_env.py:84
Methodselect_action
Gets action from model-free recovery policy Arguments: Current state Returns: ac
recovery_rl/qrisk.py:198
Methodset_goal_obj_pose
(self, pose)
env/base_mujoco_env.py:34
Methodset_y_range
(self, bounds)
env/obj_dynamic_extraction.py:189
Methodset_y_range
(self, bounds)
env/obj_extraction.py:150
Methodsetup
(self, cost_function)
recovery_rl/optimizers.py:18
Functionshuffle_rows
(arr)
recovery_rl/VisualMPC.py:55
Methodsnapshot_noarm
(self)
env/base_mujoco_env.py:153
Functionsoft_update
(target, source, tau)
recovery_rl/model.py:30
Methodstep
(self, action)
env/obj_extraction.py:91
Methodstep
(self, a)
env/navigation1.py:71
Methodstep
(self, a)
env/navigation2.py:70
Methodtarg_proc
(obs, next_obs)
config/image_maze.py:138
Methodtarg_proc
(obs, next_obs)
config/obj_dynamic_extraction.py:138
Methodtarg_proc
(obs, next_obs)
config/obj_extraction.py:138
Methodtarg_proc
(obs, next_obs)
config/navigation1.py:136
Methodtarg_proc
(obs, next_obs)
config/navigation2.py:136
Methodtarget_object_height
(self)
env/obj_dynamic_extraction.py:269
Methodtarget_object_height
(self)
env/obj_extraction.py:224
Methodto
(self, device)
recovery_rl/model.py:440
Methodto
(self, device)
recovery_rl/model.py:481
Methodto
(self, device)
recovery_rl/model.py:527
Methodto
(self, device)
recovery_rl/model.py:617
Methodto
(self, device)
recovery_rl/model.py:704
Methodtrain
Trains this controller using lists of trajectories.
recovery_rl/MPC.py:29
← previousnext →201–300 of 314, ranked by callers