MCPcopy Create free account

hub / github.com/aviralkumar2907/CQL / functions

Functions729 in github.com/aviralkumar2907/CQL

↓ 1 callersMethodcollect_one_step
( self, max_path_length, discard_incomplete_paths, )
d4rl/rlkit/samplers/data_collector/step_collector.py:67
↓ 1 callersMethodcollect_one_step
( self, max_path_length, discard_incomplete_paths, )
d4rl/rlkit/samplers/data_collector/step_collector.py:193
↓ 1 callersFunctioncreate_exp_name
Create a semi-unique experiment name that has a timestamp :param exp_prefix: :param exp_id: :return:
d4rl/rlkit/launchers/launcher_util.py:169
↓ 1 callersFunctioncreate_log_dir
Creates and returns a unique log directory. :param exp_prefix: All experiments with this prefix will have log directories be under this
d4rl/rlkit/launchers/launcher_util.py:181
↓ 1 callersFunctioncreate_mounts
( mode, base_log_dir, sync_interval=180, local_input_dir_to_mount_point_dict=N
d4rl/rlkit/launchers/launcher_util.py:817
↓ 1 callersFunctiondict_to_safe_json
Convert each value in the dictionary into a JSON'able primitive. :param d: :return:
d4rl/rlkit/launchers/launcher_util.py:306
↓ 1 callersMethoddump_uniform_imgs_and_reconstructions
(self, dataset, epoch)
d4rl/rlkit/torch/vae/vae_trainer.py:543
↓ 1 callersFunctiondump_video
( env, policy, filename, rollout_function, rows=3, columns=6,
d4rl/rlkit/util/video.py:12
↓ 1 callersFunctionenable_gpus
(gpu_str)
d4rl/examples/cql_mujoco_new.py:106
↓ 1 callersMethodenable_render
(self)
d4rl/rlkit/envs/vae_wrapper.py:325
↓ 1 callersMethodevolve_state
(self)
d4rl/rlkit/exploration_strategies/ou_strategy.py:46
↓ 1 callersFunctionexperiment
(variant)
d4rl/examples/cql_mujoco_new.py:27
↓ 1 callersFunctionexperiment
(variant)
d4rl/examples/ddpg.py:22
↓ 1 callersFunctionexperiment
(variant)
d4rl/examples/cql_antmaze_new.py:28
↓ 1 callersFunctionexperiment
(variant)
d4rl/examples/dqn_and_double_dqn.py:21
↓ 1 callersFunctionexperiment
(variant)
d4rl/examples/td3.py:24
↓ 1 callersFunctionexperiment
(variant)
d4rl/examples/sac.py:14
↓ 1 callersFunctionexperiment
(variant)
d4rl/examples/her/her_dqn_gridworld.py:31
↓ 1 callersFunctionexperiment
(variant)
d4rl/examples/her/her_sac_gym_fetch_reach.py:14
↓ 1 callersFunctionexperiment
(variant)
d4rl/examples/her/her_td3_multiworld_sawyer_reach.py:26
↓ 1 callersMethodforward
(self, input)
d4rl/rlkit/torch/conv_networks.py:94
↓ 1 callersFunctionfull_experiment_variant_preprocess
(variant)
d4rl/rlkit/launchers/skewfit_experiments.py:28
↓ 1 callersFunctionfunction_rollout
The following value for the following keys will be a 2D array, with the first dimension corresponding to the time dimension. - observati
d4rl/rlkit/samplers/rollout_functions.py:150
↓ 1 callersMethodgenerate_next_value
Return a value for the hyperparameter
d4rl/rlkit/util/hyperparameter.py:29
↓ 1 callersMethodgenerate_random_hyperparameters
(self)
d4rl/rlkit/util/hyperparameter.py:130
↓ 1 callersFunctiongenerate_vae_dataset
(variant)
d4rl/rlkit/launchers/skewfit_experiments.py:157
↓ 1 callersFunctiongenerate_vae_dataset_from_params
( env_class=None, env_kwargs=None, env_id=None, N=10000, test_p=0.9,
d4rl/rlkit/envs/goal_generation/pickup_goal_dataset.py:39
↓ 1 callersMethodget_action_from_raw_action
(self, action, t=0, **kwargs)
d4rl/rlkit/exploration_strategies/ou_strategy.py:52
↓ 1 callersMethodget_action_from_raw_action
(self, action, **kwargs)
d4rl/rlkit/exploration_strategies/base.py:17
↓ 1 callersMethodget_actions
(self, obs)
d4rl/rlkit/torch/networks.py:114
↓ 1 callersMethodget_actions
(self, obs_np, deterministic=False)
d4rl/rlkit/torch/sac/policies.py:72
↓ 1 callersFunctionget_asset_xml
(xml_name)
d4rl/rlkit/envs/mujoco_env.py:61
↓ 1 callersFunctionget_average_returns
(paths)
d4rl/rlkit/core/eval_util.py:67
↓ 1 callersMethodget_diagnostics
(self)
d4rl/rlkit/torch/sac/cql.py:410
↓ 1 callersMethodget_encoding_from_latent_distribution_params
:param latent_distribution_params: :return: get latents from latent distribution params
d4rl/rlkit/torch/vae/vae_base.py:69
↓ 1 callersFunctionget_envs
(variant)
d4rl/rlkit/launchers/skewfit_experiments.py:300
↓ 1 callersMethodget_epoch_paths
(self)
d4rl/rlkit/samplers/data_collector/step_collector.py:149
↓ 1 callersFunctionget_flat_params
(model)
d4rl/rlkit/core/batch_rl_algorithm.py:20
↓ 1 callersFunctionget_git_infos
(dirs)
d4rl/rlkit/launchers/launcher_util.py:32
↓ 1 callersMethodget_goal
(self)
d4rl/rlkit/envs/vae_wrapper.py:187
↓ 1 callersFunctionget_image
(goal, obs, recon_obs, imsize=84, pad_length=1, pad_color=255)
d4rl/rlkit/torch/skewfit/video_gen.py:24
↓ 1 callersFunctionget_image
(goal, obs, recon_obs, imsize=84, pad_length=1, pad_color=255)
d4rl/rlkit/util/video.py:96
↓ 1 callersMethodget_mp_info
(self)
d4rl/rlkit/data_management/shared_obs_dict_replay_buffer.py:108
↓ 1 callersFunctionget_stat_in_paths
(paths, dict_name, scalar_name)
d4rl/rlkit/samplers/util.py:113
↓ 1 callersMethodinit_from_mp_info
The intended use is to have a subprocess serialize/copy a SharedObsDictRelabelingBuffer instance and call init_from on the in
d4rl/rlkit/data_management/shared_obs_dict_replay_buffer.py:79
↓ 1 callersMethodinit_serialization
(self, locals)
d4rl/rlkit/envs/mujoco_env.py:54
↓ 1 callersMethodinit_vae_training_subprocess
(self)
d4rl/rlkit/torch/skewfit/online_vae_algorithm.py:121
↓ 1 callersFunctionload_hdf5
(dataset, replay_buffer)
d4rl/examples/cql_antmaze_new.py:17
↓ 1 callersMethodload_single_buffer
Load a single replay buffer.
atari/batch_rl/fixed_replay/replay_memory/fixed_replay_buffer.py:74
↓ 1 callersMethodlog_final_buffer
(self)
atari/batch_rl/baselines/agents/dqn_agent.py:47
↓ 1 callersMethodlog_loss_under_uniform
(self, model, data, priority_function_kwargs)
d4rl/rlkit/torch/vae/vae_trainer.py:505
↓ 1 callersMethodlog_variant
(self, log_file, variant_data)
d4rl/rlkit/core/logging.py:229
↓ 1 callersMethodnormalize
(self, v, clip_range=None)
d4rl/rlkit/torch/data_management/normalizer.py:12
↓ 1 callersMethodpop_prefix
(self, )
d4rl/rlkit/core/logging.py:276
↓ 1 callersMethodprint_tabular
(self, new_tabular)
d4rl/rlkit/core/logging.py:28
↓ 1 callersMethodquickFixedReplayREMFlags
Assign flags for a quick run of FixedReplay agent.
atari/batch_rl/tests/fixed_replay_runner_test.py:45
↓ 1 callersMethodquick_init
(self, locals_)
d4rl/rlkit/core/serializable.py:17
↓ 1 callersMethodrefresh
(self)
d4rl/rlkit/core/logging.py:36
↓ 1 callersMethodreload_buffer
(self, num_buffers=None)
atari/batch_rl/fixed_replay/replay_memory/fixed_replay_buffer.py:196
↓ 1 callersMethodreparameterize
:param latent_distribution_params: :return: latents
d4rl/rlkit/torch/vae/vae_base.py:35
↓ 1 callersFunctionreset_execution_environment
Call this between calls to separate experiments. :return:
d4rl/rlkit/launchers/launcher_util.py:348
↓ 1 callersMethodrsample
(self, latent_distribution_params)
d4rl/rlkit/torch/vae/vae_base.py:97
↓ 1 callersFunctionsafe_json
(data)
d4rl/rlkit/pythonplusplus.py:171
↓ 1 callersFunctionsafe_json
(data)
d4rl/rlkit/launchers/launcher_util.py:324
↓ 1 callersFunctionsave_experiment_data
(dictionary, log_dir)
d4rl/rlkit/launchers/launcher_util.py:83
↓ 1 callersMethodsave_itr_params
(self, itr, params)
d4rl/rlkit/core/logging.py:280
↓ 1 callersMethodset_goal
Assume goal contains both image_desired_goal and any goals required for wrapped envs :param goal: :return:
d4rl/rlkit/envs/vae_wrapper.py:227
↓ 1 callersMethodset_log_tabular_only
(self, log_tabular_only)
d4rl/rlkit/core/logging.py:152
↓ 1 callersMethodset_mean
(self, mean)
d4rl/rlkit/data_management/normalizer.py:98
↓ 1 callersFunctionset_seed
Set the seed for all the possible random number generators. :param seed: :return: None
d4rl/rlkit/launchers/launcher_util.py:336
↓ 1 callersMethodset_snapshot_dir
(self, dir_name)
d4rl/rlkit/core/logging.py:134
↓ 1 callersMethodset_snapshot_gap
(self, gap)
d4rl/rlkit/core/logging.py:149
↓ 1 callersMethodset_snapshot_mode
(self, mode)
d4rl/rlkit/core/logging.py:143
↓ 1 callersMethodset_std
(self, std)
d4rl/rlkit/data_management/normalizer.py:101
↓ 1 callersMethodset_vae
(self, vae)
d4rl/rlkit/torch/vae/vae_trainer.py:279
↓ 1 callersFunctionsetup_pickup_image_env
Image env and pickup env will have presampled goals. VAE wrapper should encode whatever presampled goal is sampled.
d4rl/rlkit/envs/goal_generation/pickup_goal_dataset.py:12
↓ 1 callersFunctionsimulate_policy
(args)
d4rl/scripts/run_policy.py:11
↓ 1 callersFunctionsimulate_policy
(args)
d4rl/scripts/run_goal_conditioned_policy.py:10
↓ 1 callersFunctionskewfit_experiment
(variant)
d4rl/rlkit/launchers/skewfit_experiments.py:446
↓ 1 callersFunctionskewfit_preprocess_variant
(variant)
d4rl/rlkit/launchers/skewfit_experiments.py:439
↓ 1 callersFunctionsplit_paths
Stack multiples obs/actions/etc. from different paths :param paths: List of paths, where one path is something returned from the rollout
d4rl/rlkit/samplers/util.py:76
↓ 1 callersFunctionsplit_s3_full_path
Split "s3://foo/bar/baz" into "foo" and "bar/baz"
d4rl/rlkit/util/io.py:55
↓ 1 callersFunctionstack_list
(lst)
d4rl/rlkit/data_management/path_builder.py:56
↓ 1 callersFunctionstring_tuple_to_string
(strings)
d4rl/rlkit/pythonplusplus.py:419
↓ 1 callersMethodterminate
(self)
d4rl/rlkit/envs/wrappers.py:33
↓ 1 callersMethodterminate_episode
Let the replay buffer know that the episode has terminated in case some special book-keeping has to happen. :return:
d4rl/rlkit/data_management/replay_buffer.py:18
↓ 1 callersMethodtf_get_priority
Gets the priorities for the given indices. Args: indices: tf.Tensor with dtype int32 and shape [n]. Returns: priorities: tf.Tens
atari/batch_rl/baselines/replay_memory/logged_prioritized_replay_buffer.py:151
↓ 1 callersMethodtf_set_priority
Sets the priorities for the given indices. Args: indices: tf.Tensor with dtype int32 and shape [n]. priorities: tf.Tensor with dtype
atari/batch_rl/baselines/replay_memory/logged_prioritized_replay_buffer.py:137
↓ 1 callersMethodtrain_from_torch
(self, batch)
d4rl/rlkit/torch/torch_rl_algorithm.py:48
↓ 1 callersFunctiontrain_vae
(variant, return_data=False)
d4rl/rlkit/launchers/skewfit_experiments.py:93
↓ 1 callersFunctiontrain_vae_and_update_variant
(variant)
d4rl/rlkit/launchers/skewfit_experiments.py:58
↓ 1 callersMethodtry_render
(self, obs)
d4rl/rlkit/envs/vae_wrapper.py:335
↓ 1 callersFunctionunnormalize_image
(image)
d4rl/rlkit/data_management/obs_dict_replay_buffer.py:309
↓ 1 callersMethodupdate_policy
(self, new_policy)
d4rl/rlkit/samplers/data_collector/path_collector.py:31
↓ 1 callersMethodupdate_train_weights
(self)
d4rl/rlkit/torch/vae/vae_trainer.py:241
↓ 1 callersMethodverifyFilesCreated
Verify that files have been created.
atari/batch_rl/tests/fixed_replay_runner_test.py:58
Method__getattr__
(self, attr)
d4rl/rlkit/envs/wrappers.py:37
Method__getitem__
(self, y)
d4rl/rlkit/pythonplusplus.py:371
Method__getitem__
(self, idxs)
d4rl/rlkit/torch/data.py:21
Method__init__
(self, **kwargs)
d4rl/rlkit/pythonplusplus.py:366
← previousnext →201–300 of 729, ranked by callers