MCPcopy Create free account

hub / github.com/aviralkumar2907/CQL / functions

Functions729 in github.com/aviralkumar2907/CQL

Method_build_replay_buffer
Creates the replay buffer used by the agent.
atari/batch_rl/fixed_replay/agents/quantile_agent.py:85
Method_build_replay_buffer
Creates the replay buffer used by the agent.
atari/batch_rl/fixed_replay/agents/rainbow_agent.py:83
Method_build_replay_buffer
Creates the replay buffer used by the agent.
atari/batch_rl/fixed_replay/agents/multi_network_dqn_agent.py:71
Method_build_train_op
Builds a training op. Returns: train_op: An op performing one step of training from replay data.
atari/batch_rl/multi_head/multi_head_dqn_agent.py:118
Method_build_train_op
Builds a training op. Returns: train_op: An op performing one step of training.
atari/batch_rl/multi_head/quantile_agent.py:163
Method_build_train_op
Builds a training op. Returns: train_op: An op performing one step of training from replay data.
atari/batch_rl/multi_head/multi_network_dqn_agent.py:186
Function_column_type
The least generic type all column values are convertible to. >>> _column_type(["1", "2"]) is _int_type True >>> _column_type(["1", "2.3"]
d4rl/rlkit/core/tabulate.py:407
Method_create_network
Builds a multi-head Q-network that outputs Q-values for multiple heads. Args: name: str, this name is passed to the tf.keras.Model and used
atari/batch_rl/multi_head/multi_head_dqn_agent.py:78
Method_create_network
Builds a Quantile ConvNet. Equivalent to Rainbow ConvNet, only now the output logits are interpreted as quantiles. Args: name: str
atari/batch_rl/multi_head/quantile_agent.py:122
Method_create_network
Builds a multi-network Q-network that outputs Q-values for each network. Args: name: str, this name is passed to the tf.keras.Model and use
atari/batch_rl/multi_head/multi_network_dqn_agent.py:86
Method_dump_imgs_and_reconstructions
(self, idxs, filename)
d4rl/rlkit/torch/vae/vae_trainer.py:485
Method_end_epoch
(self, epoch)
d4rl/rlkit/torch/skewfit/online_vae_algorithm.py:52
Method_get_obs
(self, history_flat, true_state)
d4rl/rlkit/envs/mujoco_image_env.py:146
Method_get_snapshot
(self)
d4rl/rlkit/torch/skewfit/online_vae_algorithm.py:65
Method_get_snapshot
(self)
d4rl/rlkit/core/rl_algorithm.py:69
Method_get_sorted_idx_and_train_weights
(self)
d4rl/rlkit/data_management/online_vae_replay_buffer.py:306
Method_initialize_checkpointer_and_maybe_resume
(self, checkpoint_file_prefix)
atari/batch_rl/fixed_replay/run_experiment.py:36
Function_latex_line_begin_tabular
(colwidths, colaligns)
d4rl/rlkit/core/tabulate.py:112
Method_load_and_save_buffers
(self, num_load=None, num_buffers=None)
atari/batch_rl/fixed_replay/replay_memory/fixed_replay_buffer.py:136
Method_load_replay_buffers_initial
Load multiple checkpoints into a list of replay buffers.
atari/batch_rl/fixed_replay/replay_memory/fixed_replay_buffer.py:100
Method_log_stats
(self, epoch)
d4rl/rlkit/torch/skewfit/online_vae_algorithm.py:57
Function_mediawiki_row_with_attrs
(separator, cell_values, colwidths, colaligns)
d4rl/rlkit/core/tabulate.py:99
Function_more_generic
(type1, type2)
d4rl/rlkit/core/tabulate.py:400
Function_pipe_line_with_colons
Return a horizontal line with optional colons to indicate column's alignment (as in `pipe` output format).
d4rl/rlkit/core/tabulate.py:92
Method_run_one_iteration
Runs one iteration of agent/environment interaction.
atari/batch_rl/fixed_replay/run_experiment.py:73
Method_sample_goals_from_env
(self, batch_size)
d4rl/rlkit/data_management/online_vae_replay_buffer.py:252
Method_size
(self)
d4rl/rlkit/data_management/shared_obs_dict_replay_buffer.py:117
Method_train
(self)
d4rl/rlkit/torch/skewfit/online_vae_algorithm.py:48
Method_train
(self)
d4rl/rlkit/core/online_rl_algorithm.py:50
Method_train
(self)
d4rl/rlkit/core/batch_rl_algorithm.py:108
Function_visible_width
Visible width of a printed string. ANSI color codes are removed. >>> _visible_width('\x1b[31mhello\x1b[0m'), _visible_width("world") (5, 5)
d4rl/rlkit/core/tabulate.py:349
Method_visualize
(self, policy=False, q_function=False, num_dir=50, alpha=0.1, iter=None)
d4rl/rlkit/core/batch_rl_algorithm.py:244
Methodadd
(self, observation, action, reward, terminal, *args)
atari/batch_rl/baselines/replay_memory/logged_prioritized_replay_buffer.py:51
Methodadd
(self, observation, action, reward, terminal, *args)
atari/batch_rl/baselines/replay_memory/logged_replay_buffer.py:47
Methodadd_path
(self, path)
d4rl/rlkit/data_management/online_vae_replay_buffer.py:91
Methodadd_path
(self, path)
d4rl/rlkit/data_management/obs_dict_replay_buffer.py:100
Methodadd_sample
(self, observation, action, reward, next_observation, terminal, env_info, **kwargs)
d4rl/rlkit/data_management/simple_replay_buffer.py:41
Methodadd_sample
(self, observation, action, reward, terminal, next_observation, **kwargs)
d4rl/rlkit/data_management/env_replay_buffer.py:36
Methodadd_sample
(self, observation, action, reward, terminal, next_observation, **kwargs)
d4rl/rlkit/data_management/obs_dict_replay_buffer.py:90
Methodalways_false
(self)
d4rl/rlkit/pythonplusplus.py:415
Functionalways_train
(epoch)
d4rl/rlkit/torch/vae/vae_schedules.py:1
Functionbatch
Split an interable into batches of size `n`. If `n` does not evenly divide `iterable`, the last slice will be smaller. https://stackover
d4rl/rlkit/pythonplusplus.py:287
Methodbatch_env_info_dict
(self, indices)
d4rl/rlkit/data_management/simple_replay_buffer.py:81
Methodbegin_episode
Returns the agent's first action for this episode. Args: observation: numpy array, the environment's initial observation. Returns:
atari/batch_rl/multi_head/multi_network_dqn_agent.py:127
Methodcall
Calculates the distribution of Q-values using the input state tensor.
atari/batch_rl/multi_head/atari_helpers.py:94
Methodcall
Creates the output tensor/op given the input state tensor. See https://www.tensorflow.org/api_docs/python/tf/keras/Model for more information
atari/batch_rl/multi_head/atari_helpers.py:185
Methodcall
Creates the output tensor/op given the state tensor as input. See https://www.tensorflow.org/api_docs/python/tf/keras/Model for more informat
atari/batch_rl/multi_head/atari_helpers.py:294
Methodcall
Creates the output tensor/op given the input state tensor. See https://www.tensorflow.org/api_docs/python/tf/keras/Model for more information
atari/batch_rl/multi_head/atari_helpers.py:361
Methodcheck
(self, time)
d4rl/rlkit/pythonplusplus.py:404
Functionclip_magnitude
Clip the magnitude of value to be within some value. :param value: :param magnitude: :return:
d4rl/rlkit/pythonplusplus.py:16
Methodclone
(cls, obj, **kwargs)
d4rl/rlkit/core/serializable.py:59
Methodcollect_new_paths
(self, *args, **kwargs)
d4rl/rlkit/samplers/data_collector/vae_env.py:18
Methodcollect_new_paths
( self, max_path_length, num_steps, discard_incomplete_paths,
d4rl/rlkit/samplers/data_collector/path_collector.py:34
Methodcollect_new_paths
( self, policy_fn, max_path_length, num_steps, discard_incomplete_paths )
d4rl/rlkit/samplers/data_collector/path_collector.py:117
Methodcollect_new_paths
( self, max_path_length, num_steps, discard_incomplete_paths,
d4rl/rlkit/samplers/data_collector/path_collector.py:196
Methodcollect_new_steps
( self, max_path_length, num_steps, discard_incomplete_paths,
d4rl/rlkit/samplers/data_collector/step_collector.py:58
Methodcollect_new_steps
( self, max_path_length, num_steps, discard_incomplete_paths,
d4rl/rlkit/samplers/data_collector/step_collector.py:184
Functioncompute_bernoulli_log_prob
(x, reconstruction_of_x)
d4rl/rlkit/torch/vae/vae_base.py:118
Functioncompute_gaussian_log_prob
(input, dec_mu, dec_var)
d4rl/rlkit/torch/vae/vae_base.py:126
Functioncopy_model_params_from_to
(source, target)
d4rl/rlkit/torch/pytorch_util.py:12
Methodcopy_stats
(self, other)
d4rl/rlkit/data_management/normalizer.py:121
Functioncreate_agent
Creates a DQN agent. Args: sess: A `tf.Session`object for running associated ops. environment: An Atari 2600 environment. replay_log_d
atari/batch_rl/baselines/train.py:44
Functioncreate_agent
Creates a DQN agent. Args: sess: A `tf.Session`object for running associated ops. environment: An Atari 2600 environment. replay_data_
atari/batch_rl/fixed_replay/train.py:52
Functioncreate_agent
Creates an online agent. Args: sess: A `tf.Session`object for running associated ops. environment: An Atari 2600 environment. summary_
atari/online/train.py:39
Functioncustom_schedule
(epoch)
d4rl/rlkit/torch/vae/vae_schedules.py:5
Functioncustom_schedule_2
(epoch)
d4rl/rlkit/torch/vae/vae_schedules.py:14
Methoddebug_statistics
Given an image $$x$$, samples a bunch of latents from the prior $$z_i$$ and decode them $$\hat x_i$$. Compare this to $$\hat
d4rl/rlkit/torch/vae/vae_trainer.py:435
Methoddecode
(self, latents)
d4rl/rlkit/torch/vae/conv_vae.py:216
Methoddecode_goals
(self)
d4rl/rlkit/envs/vae_wrapper.py:294
Methoddefault
(self, o)
d4rl/rlkit/core/logging.py:46
Methoddenormalize
(self, v)
d4rl/rlkit/torch/data_management/normalizer.py:25
Methoddenormalize
(self, v)
d4rl/rlkit/torch/data_management/normalizer.py:57
Methoddenormalize
(self, v)
d4rl/rlkit/data_management/normalizer.py:47
Methoddenormalize
(self, v)
d4rl/rlkit/data_management/normalizer.py:77
Methoddenormalize
(self, v)
d4rl/rlkit/data_management/normalizer.py:114
Methoddenormalize_scale
Only denormalize the scale. Do not add the mean.
d4rl/rlkit/torch/data_management/normalizer.py:65
Functiondict_of_list__to__list_of_dicts
``` x = {'foo': [3, 4, 5], 'bar': [1, 2, 3]} ppp.dict_of_list__to__list_of_dicts(x, 3) # Output: # [ # {'foo': 3, 'bar':
d4rl/rlkit/pythonplusplus.py:124
Functiondict_to_safe_json
Convert each value in the dictionary into a JSON'able primitive. :param d: :return:
d4rl/rlkit/pythonplusplus.py:183
Methoddisable_render
(self)
d4rl/rlkit/envs/vae_wrapper.py:330
Functiondot_map_dict_to_nested_dict
Convert something like ``` { 'one.two.three.four': 4, 'one.six.seven.eight': None, 'five.nine.ten': 10, '
d4rl/rlkit/pythonplusplus.py:38
Functiondump_video
( env, policy, filename, rollout_function, rows=3, columns=6,
d4rl/rlkit/torch/skewfit/video_gen.py:36
Functionenable_gpus
(gpu_str)
d4rl/examples/cql_antmaze_new.py:105
Methodenable_render
(self)
d4rl/rlkit/envs/mujoco_image_env.py:133
Methodencode
(self, input)
d4rl/rlkit/torch/vae/conv_vae.py:207
Methodend_collection
(self)
d4rl/rlkit/samplers/data_collector/step_collector.py:180
Methodend_episode
(self, reward)
atari/batch_rl/fixed_replay/agents/multi_head_dqn_agent.py:70
Methodend_episode
(self, reward)
atari/batch_rl/fixed_replay/agents/dqn_agent.py:79
Methodend_episode
(self, reward)
atari/batch_rl/fixed_replay/agents/quantile_agent.py:81
Methodend_episode
(self, reward)
atari/batch_rl/fixed_replay/agents/rainbow_agent.py:79
Methodend_episode
(self, reward)
atari/batch_rl/fixed_replay/agents/multi_network_dqn_agent.py:67
Methodend_epoch
(self, epoch)
d4rl/rlkit/torch/sac/cql.py:413
Methodend_epoch
(self, epoch)
d4rl/rlkit/torch/sac/sac.py:206
Methodend_epoch
(self, epoch)
d4rl/rlkit/torch/ddpg/ddpg.py:190
Methodend_epoch
(self, epoch)
d4rl/rlkit/torch/td3/td3.py:175
Methodend_epoch
(self, epoch)
d4rl/rlkit/torch/her/her.py:22
Methodend_epoch
(self, epoch)
d4rl/rlkit/torch/dqn/dqn.py:92
Methodend_epoch
(self, epoch)
d4rl/rlkit/data_management/replay_buffer.py:92
Methodend_epoch
(self, epoch)
d4rl/rlkit/samplers/data_collector/step_collector.py:35
Methodend_epoch
(self, epoch)
d4rl/rlkit/samplers/data_collector/step_collector.py:152
Methodend_epoch
(self, epoch)
d4rl/rlkit/samplers/data_collector/base.py:5
← previousnext →401–500 of 729, ranked by callers