MCPcopy Create free account

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

Functions314 in github.com/abalakrishna123/recovery-rl

↓ 87 callersMethodto
(self, device)
recovery_rl/model.py:340
↓ 24 callersFunctionget_affine_params
(ensemble_size, in_features, out_features)
config/utils.py:15
↓ 18 callersFunctionswish
(x)
config/utils.py:6
↓ 16 callersMethodsample
(self, state)
recovery_rl/model.py:325
↓ 14 callersFunctionget_required_argument
(dotmap, key, message, default=None)
recovery_rl/utils.py:72
↓ 12 callersMethod__init__
(self, num_inputs, num_actions, hidden_dim)
recovery_rl/model.py:50
↓ 12 callersMethodplot
Interface for visualizing Q_risk for all navigation environments.
recovery_rl/qrisk.py:229
↓ 12 callersMethodstep
Applies the action and steps simulation :param action: action at time-step :return: obs dict where: -each k
env/base_env.py:13
↓ 10 callersMethodrender
Renders the enviornment. Implements custom rendering support. If mode is: - dual: renders both left and main cameras - left:
env/base_env.py:64
↓ 10 callersMethodreset
Resets this controller (clears previous solution, calls all update functions). Returns: None
recovery_rl/MPC.py:311
↓ 9 callersFunctionget_color
(algname, alt_color_map={})
plotting/plotting_utils.py:26
↓ 8 callersMethod_get_obs
(self, images=False)
env/maze.py:170
↓ 7 callersMethodpush
(self, state, action, reward, next_state, done)
recovery_rl/replay_memory.py:21
↓ 6 callersMethod_next_state
(self, s, a, override=False)
env/navigation2.py:98
↓ 6 callersMethodstep_cost
(self, s, a)
env/navigation2.py:105
↓ 5 callersFunctionget_legend_name
(algname, alt_name_map={})
plotting/plotting_utils.py:35
↓ 5 callersMethodget_value
Arguments: states, actions --> list of states and list of corresponding actions to get Q_risk values for
recovery_rl/qrisk.py:184
↓ 5 callersMethodseed
(self, seed=None)
env/base_env.py:90
↓ 5 callersMethodupdate_parameters
Train task policy and associated Q function with experience in replay buffer (memory)
recovery_rl/sac.py:170
↓ 4 callersMethod_get_obs
(self, images=False)
env/image_maze.py:200
↓ 4 callersMethodforward
(self, inputs, ret_logvar=False)
config/maze.py:71
↓ 4 callersFunctionget_stats
(data)
plotting/plot_runs.py:14
↓ 4 callersFunctionprocess_obs
(obs)
env/image_maze.py:29
↓ 4 callersFunctionprocess_obs
(obs)
env/maze.py:29
↓ 4 callersFunctionprocess_obs
(obs, env_name)
recovery_rl/experiment.py:35
↓ 4 callersFunctiontorchify
(x)
recovery_rl/experiment.py:26
↓ 3 callersMethod_reset_sim
Creates a MjSim from passed in model_path :param model_path: Absolute path to model file :return: None
env/base_mujoco_env.py:37
↓ 3 callersMethodget_distance_score
(self)
env/image_maze.py:246
↓ 3 callersMethodget_distance_score
:return: mean of the distances between all objects and goals
env/maze.py:215
↓ 3 callersMethodsample
(self, batch_size)
recovery_rl/replay_memory.py:27
↓ 3 callersMethodselect_action
Get action from current task policy
recovery_rl/sac.py:133
↓ 3 callersMethodtarg_proc
(obs, next_obs)
config/maze.py:138
↓ 3 callersMethodtrain
Trains the internal model of this controller. Once trained, this controller switches from applying random actions to using MPC. Argum
recovery_rl/MPC.py:213
↓ 2 callersMethod_expand_to_ts_format
(self, mat)
recovery_rl/MPC.py:441
↓ 2 callersMethod_next_state
(self, s, a, override=False)
env/navigation1.py:99
↓ 2 callersFunctionclip_target_qpos
(target, lb, ub)
env/obj_dynamic_extraction.py:31
↓ 2 callersMethodget_action
(self, state, train=True)
recovery_rl/experiment.py:546
↓ 2 callersFunctionhard_update
(target, source)
recovery_rl/utils.py:52
↓ 2 callersFunctionlinear_schedule
(startval, endval, endtime)
recovery_rl/utils.py:62
↓ 2 callersFunctionmake_env
(env_name)
env/make_utils.py:29
↓ 2 callersFunctionno_rot_dynamics
(prev_target_qpos, action)
env/obj_dynamic_extraction.py:24
↓ 2 callersMethodproject_point
(self, point, camera)
env/base_mujoco_env.py:68
↓ 2 callersFunctionregister_env
(env_name)
env/make_utils.py:23
↓ 2 callersMethodrender
(self)
env/obj_dynamic_extraction.py:67
↓ 2 callersMethodrender
(self)
env/obj_extraction.py:64
↓ 2 callersMethodreset
(self, difficulty='m', check_constraint=True, pos=())
env/image_maze.py:221
↓ 2 callersMethodreset
(self, difficulty='h', check_constraint=True, pos=())
env/maze.py:184
↓ 2 callersMethodsample
samples a random action from the action space.
env/navigation1.py:115
↓ 2 callersFunctionsoft_update
(target, source, tau)
recovery_rl/utils.py:46
↓ 2 callersMethodstep
(self, action)
env/image_maze.py:169
↓ 2 callersMethodstep
(self, action)
env/maze.py:139
↓ 2 callersMethodstep_cost
(self, s, a)
env/navigation1.py:106
↓ 2 callersMethodtrain_MB_recovery
(self, states, actions, next_states=None, epochs=50)
recovery_rl/experiment.py:251
↓ 1 callersMethod__init__
(self, *args, **kwargs)
recovery_rl/optimizers.py:15
↓ 1 callersFunction_create_ctrl_config
Creates controller configuration.
config/default.py:64
↓ 1 callersFunction_create_exp_config
(exp_cfg, cfg_module, logdir, type_map)
config/default.py:54
↓ 1 callersMethod_flatten_to_matrix
(self, ts_fmt_arr)
recovery_rl/MPC.py:457
↓ 1 callersMethod_next_qpos
(self, action)
env/obj_dynamic_extraction.py:272
↓ 1 callersMethod_next_qpos
(self, action)
env/obj_extraction.py:227
↓ 1 callersMethod_next_qpos_dynamic_obs
(self, action)
env/obj_dynamic_extraction.py:278
↓ 1 callersMethod_predict_next_obs
(self, obs, acs)
recovery_rl/MPC.py:421
↓ 1 callersMethodact
Returns the action that this controller would take at time t given observation obs. Arguments: obs: The current observation
recovery_rl/MPC.py:322
↓ 1 callersMethodagent_setup
(self, env)
recovery_rl/experiment.py:169
↓ 1 callersFunctionclip_target_qpos
(target, lb, ub)
env/obj_extraction.py:29
↓ 1 callersMethodcompute_decays
(self)
config/maze.py:52
↓ 1 callersFunctioncreate_conditional
(cl, cond, message)
config/default.py:136
↓ 1 callersFunctioncreate_config
(env_name, ctrl_type, ctrl_args, overrides, logdir)
config/default.py:15
↓ 1 callersMethoddump_logs
(self, train_rollouts, test_rollouts)
recovery_rl/experiment.py:540
↓ 1 callersMethodexperiment_setup
(self)
recovery_rl/experiment.py:89
↓ 1 callersMethodexpert_action
(self)
env/image_maze.py:250
↓ 1 callersMethodexpert_action
(self)
env/maze.py:222
↓ 1 callersMethodfit_input_stats
(self, data)
config/maze.py:61
↓ 1 callersMethodforward
(self, state)
recovery_rl/model.py:317
↓ 1 callersMethodforward
(self, state)
recovery_rl/model.py:407
↓ 1 callersMethodforward
(self, state)
recovery_rl/model.py:468
↓ 1 callersMethodforward
(self, state)
recovery_rl/model.py:512
↓ 1 callersMethodforward
(self, state)
recovery_rl/model.py:594
↓ 1 callersMethodforward
(self, state)
recovery_rl/model.py:680
↓ 1 callersFunctionget_args
()
arg_utils.py:8
↓ 1 callersMethodget_contact_info
(self)
env/obj_dynamic_extraction.py:154
↓ 1 callersMethodget_distance_score
:return: mean of the distances between all objects and goals
env/base_mujoco_env.py:140
↓ 1 callersFunctionget_img_from_fig
(fig, dpi=180)
env/navigation1.py:168
↓ 1 callersFunctionget_img_from_fig
(fig, dpi=180)
env/navigation2.py:247
↓ 1 callersMethodget_offline_data
(self)
recovery_rl/experiment.py:177
↓ 1 callersMethodget_test_rollout
(self, i_episode)
recovery_rl/experiment.py:493
↓ 1 callersMethodget_train_rollout
(self, i_episode)
recovery_rl/experiment.py:379
↓ 1 callersFunctionmoving_average
(x, N)
plotting/plot_runs.py:23
↓ 1 callersFunctionno_rot_dynamics
(prev_target_qpos, action)
env/obj_extraction.py:22
↓ 1 callersFunctionnpy_to_gif
(im_list, filename, fps=4)
recovery_rl/experiment.py:29
↓ 1 callersFunctionnpy_to_gif
(im_list, filename, fps=4)
recovery_rl/VisualMPC.py:60
↓ 1 callersMethodobject_reset_poses
(self)
env/obj_dynamic_extraction.py:232
↓ 1 callersMethodobject_reset_poses
(self)
env/obj_extraction.py:187
↓ 1 callersMethodobs_postproc
(obs, pred)
config/maze.py:134
↓ 1 callersMethodobtain_solution
(self, *args, **kwargs)
recovery_rl/optimizers.py:24
↓ 1 callersFunctionplot_experiment
Construct experiment map for this experiment
plotting/plot_runs.py:141
↓ 1 callersMethodpretrain_critic_recovery
(self)
recovery_rl/experiment.py:261
↓ 1 callersMethodpretrain_task_critic
(self)
recovery_rl/experiment.py:337
↓ 1 callersFunctionprocess_action
(a)
env/image_maze.py:25
↓ 1 callersFunctionprocess_action
(a)
env/navigation1.py:50
↓ 1 callersFunctionprocess_action
(a)
env/maze.py:25
next →1–100 of 314, ranked by callers