MCPcopy Create free account

hub / github.com/aviralkumar2907/CQL / functions

Functions729 in github.com/aviralkumar2907/CQL

↓ 2 callersMethoddump_samples
(self, epoch)
d4rl/rlkit/torch/vae/vae_trainer.py:475
↓ 2 callersFunctioneval_np
Eval this module with a numpy interface Same as a call to __call__ except all Variable input/outputs are replaced with numpy equivalents
d4rl/rlkit/torch/core.py:7
↓ 2 callersFunctionflatten_n
(xs)
d4rl/rlkit/data_management/obs_dict_replay_buffer.py:269
↓ 2 callersMethodforward
(self, input, return_preactivations=False)
d4rl/rlkit/torch/networks.py:66
↓ 2 callersMethodget_action
(self, t, observation, policy, **kwargs)
d4rl/rlkit/exploration_strategies/base.py:8
↓ 2 callersMethodget_all_stacked
(self)
d4rl/rlkit/data_management/path_builder.py:46
↓ 2 callersFunctionget_dim
(space)
d4rl/rlkit/envs/env_utils.py:12
↓ 2 callersFunctionload_hdf5
(dataset, replay_buffer)
d4rl/examples/cql_mujoco_new.py:17
↓ 2 callersFunctionlocal_path_from_s3_or_local_path
(filename)
d4rl/rlkit/util/io.py:15
↓ 2 callersMethodlog_diagnostics
(self, paths)
d4rl/rlkit/envs/mujoco_env.py:57
↓ 2 callersMethodlog_prob
:param value: some value, x :param pre_tanh_value: arctanh(x) :return:
d4rl/rlkit/torch/distributions.py:32
↓ 2 callersMethodlogprob
:param inputs: :param obs_distribution_params: :return: log probability of input under decoder
d4rl/rlkit/torch/vae/vae_base.py:52
↓ 2 callersFunctionmkdir_p
(path)
d4rl/rlkit/core/logging.py:60
↓ 2 callersFunctionmultitask_rollout
( env, agent, max_path_length=np.inf, render=False, render_kwargs=None
d4rl/rlkit/samplers/rollout_functions.py:4
↓ 2 callersFunctionnp_ify
(tensor_or_other)
d4rl/rlkit/torch/core.py:32
↓ 2 callersFunctionnp_to_pytorch_batch
(np_batch)
d4rl/rlkit/torch/core.py:55
↓ 2 callersMethodpop_tabular_prefix
(self, )
d4rl/rlkit/core/logging.py:189
↓ 2 callersFunctionpostprocess_obs_dict
Undo internal replay buffer representation changes: save images as bytes
d4rl/rlkit/data_management/obs_dict_replay_buffer.py:294
↓ 2 callersMethodpush_prefix
(self, prefix)
d4rl/rlkit/core/logging.py:110
↓ 2 callersMethodpush_tabular_prefix
(self, key)
d4rl/rlkit/core/logging.py:185
↓ 2 callersFunctionquery_yes_no
Ask a yes/no question via raw_input() and return their answer. "question" is a string that is presented to the user. "default" is the presume
d4rl/rlkit/launchers/launcher_util.py:356
↓ 2 callersMethodrefresh_latents
(self, epoch)
d4rl/rlkit/data_management/online_vae_replay_buffer.py:136
↓ 2 callersFunctionrelative_probs_from_log_probs
Returns relative probability from the log probabilities. They're not exactly equal to the probability, but relative scalings between them are
d4rl/rlkit/torch/vae/vae_trainer.py:21
↓ 2 callersMethodremove_tabular_output
(self, file_name, relative_to_snapshot_dir=False)
d4rl/rlkit/core/logging.py:127
↓ 2 callersMethodreset
(self)
d4rl/rlkit/exploration_strategies/base.py:11
↓ 2 callersMethodreset
(self, **kwargs)
d4rl/rlkit/envs/wrappers.py:20
↓ 2 callersFunctionrollout
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:76
↓ 2 callersMethodrsample
Sampling in the reparameterization case.
d4rl/rlkit/torch/distributions.py:60
↓ 2 callersMethodsample_goals
(self, batch_size)
d4rl/rlkit/envs/vae_wrapper.py:140
↓ 2 callersMethodsample_weighted_indices
(self, batch_size)
d4rl/rlkit/data_management/online_vae_replay_buffer.py:233
↓ 2 callersFunctionset_flat_params
(model, flat_params, trainable_only=True)
d4rl/rlkit/core/batch_rl_algorithm.py:27
↓ 2 callersFunctionset_gpu_mode
(mode, gpu_id=0)
d4rl/rlkit/torch/pytorch_util.py:52
↓ 2 callersFunctionsync_down
(path, check_exists=True)
d4rl/rlkit/util/io.py:25
↓ 2 callersFunctiontabulate
Format a fixed width table for pretty printing. >>> print(tabulate([[1, 2.34], [-56, "8.999"], ["2", "10001"]])) --- --------- 1
d4rl/rlkit/core/tabulate.py:541
↓ 2 callersFunctiontemporary_mode
(env, mode, func, args=None, kwargs=None)
d4rl/rlkit/envs/vae_wrapper.py:422
↓ 2 callersMethodtest_epoch
( self, epoch, save_reconstruction=True, save_vae=True,
d4rl/rlkit/torch/vae/vae_trainer.py:375
↓ 2 callersFunctiontorch_ify
(np_array_or_other)
d4rl/rlkit/torch/core.py:25
↓ 2 callersMethodtrain_epoch
(self, epoch, sample_batch=None, batches=100, from_rl=False)
d4rl/rlkit/torch/vae/vae_trainer.py:320
↓ 1 callersFunctionFloatTensor
(*args, torch_device=None, **kwargs)
d4rl/rlkit/torch/pytorch_util.py:70
↓ 1 callersMethod__getstate__
(self)
d4rl/rlkit/core/serializable.py:45
↓ 1 callersMethod__getstate__
This is useful to override in case the wrapped env has some funky __getstate__ that doesn't play well with overriding __getattr__.
d4rl/rlkit/envs/wrappers.py:42
↓ 1 callersMethod__getstate__
(self)
d4rl/rlkit/envs/vae_wrapper.py:410
↓ 1 callersMethod__init__
(self)
d4rl/rlkit/pythonplusplus.py:430
↓ 1 callersMethod__init__
( self, input_width, input_height, input_channels,
d4rl/rlkit/torch/conv_networks.py:10
↓ 1 callersMethod__init__
(self, delta=1)
d4rl/rlkit/torch/modules.py:9
↓ 1 callersMethod__init__
( self, representation_size, )
d4rl/rlkit/torch/vae/vae_base.py:10
↓ 1 callersMethod__init__
(self, stochastic_policy)
d4rl/rlkit/torch/sac/policies.py:153
↓ 1 callersMethod__init__
(self)
d4rl/rlkit/core/logging.py:71
↓ 1 callersMethod__init__
(self, wrapped_env, imsize=32, keep_prev=0,
d4rl/rlkit/envs/mujoco_image_env.py:14
↓ 1 callersMethod__init__
Initializes WrappedLoggedPrioritizedReplayBuffer.
atari/batch_rl/baselines/replay_memory/logged_prioritized_replay_buffer.py:98
↓ 1 callersMethod__init__
Initializes WrappedLoggedReplayBuffer.
atari/batch_rl/baselines/replay_memory/logged_replay_buffer.py:97
↓ 1 callersMethod__init__
Initialize the FixedReplayBuffer class. Args: data_dir: str, log Directory from which to load the replay buffer. replay_suffix: int,
atari/batch_rl/fixed_replay/replay_memory/fixed_replay_buffer.py:39
↓ 1 callersMethod__setstate__
(self, d)
d4rl/rlkit/core/serializable.py:48
↓ 1 callersMethod__setstate__
(self, state)
d4rl/rlkit/envs/wrappers.py:52
↓ 1 callersMethod__setstate__
(self, state)
d4rl/rlkit/envs/vae_wrapper.py:417
↓ 1 callersMethod_advance
(self)
d4rl/rlkit/data_management/simple_replay_buffer.py:56
↓ 1 callersFunction_afterpoint
Symbols after a decimal point, -1 if the string lacks the decimal point. >>> _afterpoint("123.45") 2 >>> _afterpoint("1001") -1 >
d4rl/rlkit/core/tabulate.py:278
↓ 1 callersFunction_align_column
[string] -> [padded_string] >>> list(map(str,_align_column(["12.345", "-1234.5", "1.23", "1234.5", "1e+234", "1.0e234"], "decimal"))) [' 12
d4rl/rlkit/core/tabulate.py:362
↓ 1 callersFunction_align_header
(header, alignment, width)
d4rl/rlkit/core/tabulate.py:456
↓ 1 callersMethod_apply_normalize_obs
(self, obs)
d4rl/rlkit/envs/wrappers.py:152
↓ 1 callersMethod_batch_next_obs_dict
(self, indices)
d4rl/rlkit/data_management/obs_dict_replay_buffer.py:262
↓ 1 callersMethod_batch_obs_dict
(self, indices)
d4rl/rlkit/data_management/obs_dict_replay_buffer.py:256
↓ 1 callersMethod_build_target_distribution
(self)
atari/batch_rl/multi_head/quantile_agent.py:138
↓ 1 callersMethod_build_target_q_op
Build an op used as a target for the Q-value. Returns: target_q_op: An op calculating the Q-value.
atari/batch_rl/multi_head/multi_head_dqn_agent.py:103
↓ 1 callersMethod_build_target_q_op
Build an op used as a target for the Q-value. Returns: target_q_op: An op calculating the Q-value.
atari/batch_rl/multi_head/multi_network_dqn_agent.py:112
↓ 1 callersMethod_cleanup
(self)
d4rl/rlkit/torch/skewfit/online_vae_algorithm.py:116
↓ 1 callersFunction_elem_or_tuple_to_variable
(elem_or_tuple)
d4rl/rlkit/torch/core.py:39
↓ 1 callersMethod_encode_one
(self, img)
d4rl/rlkit/envs/vae_wrapper.py:382
↓ 1 callersFunction_filter_batch
(np_batch)
d4rl/rlkit/torch/core.py:47
↓ 1 callersFunction_format
Format a value accoding to its type. Unicode is supported: >>> hrow = ['\u0431\u0443\u043a\u0432\u0430', '\u0446\u0438\u0444\u0440\u0430'] ;
d4rl/rlkit/core/tabulate.py:431
↓ 1 callersFunction_format_table
Produce a plain-text representation of the table.
d4rl/rlkit/core/tabulate.py:817
↓ 1 callersFunction_get_epoch_timings
()
d4rl/rlkit/core/rl_algorithm.py:11
↓ 1 callersMethod_handle_rollout_ending
( self, max_path_length, discard_incomplete_paths )
d4rl/rlkit/samplers/data_collector/step_collector.py:104
↓ 1 callersMethod_handle_rollout_ending
( self, max_path_length, discard_incomplete_paths )
d4rl/rlkit/samplers/data_collector/step_collector.py:234
↓ 1 callersMethod_image_and_proprio_from_decoded
(self, decoded)
d4rl/rlkit/envs/vae_wrapper.py:398
↓ 1 callersMethod_load_buffer
Loads a OutOfGraphReplayBuffer replay buffer.
atari/batch_rl/fixed_replay/replay_memory/fixed_replay_buffer.py:84
↓ 1 callersMethod_load_replay_buffers
Loads multiple checkpoints into a list of replay buffers.
atari/batch_rl/fixed_replay/replay_memory/fixed_replay_buffer.py:155
↓ 1 callersMethod_log_stats
(self, epoch)
d4rl/rlkit/core/rl_algorithm.py:81
↓ 1 callersMethod_log_vae_stats
(self)
d4rl/rlkit/torch/skewfit/online_vae_algorithm.py:110
↓ 1 callersFunction_normalize_tabular_data
Transform a supported data type to a list of lists, and a list of headers. Supported tabular data types: * list-of-lists or another iterable
d4rl/rlkit/core/tabulate.py:467
↓ 1 callersFunction_padboth
Center string. >>> _padboth(6, '\u044f\u0439\u0446\u0430') == ' \u044f\u0439\u0446\u0430 ' True
d4rl/rlkit/core/tabulate.py:329
↓ 1 callersFunction_padleft
Flush right. >>> _padleft(6, '\u044f\u0439\u0446\u0430') == ' \u044f\u0439\u0446\u0430' True
d4rl/rlkit/core/tabulate.py:305
↓ 1 callersFunction_padright
Flush left. >>> _padright(6, '\u044f\u0439\u0446\u0430') == '\u044f\u0439\u0446\u0430 ' True
d4rl/rlkit/core/tabulate.py:317
↓ 1 callersFunction_pipe_segment_with_colons
Return a segment of a horizontal line with optional colons which indicate column's alignment (as in `pipe` output format).
d4rl/rlkit/core/tabulate.py:78
↓ 1 callersMethod_run_train_phase
Run training phase.
atari/batch_rl/fixed_replay/run_experiment.py:63
↓ 1 callersMethod_sample_vae_prior
(self, batch_size)
d4rl/rlkit/envs/vae_wrapper.py:369
↓ 1 callersMethod_save_tensorboard_summaries
Save statistics as tensorboard summaries. Args: iteration: int, The current iteration number. num_episodes_eval: int, number of evalu
atari/batch_rl/fixed_replay/run_experiment.py:90
↓ 1 callersMethod_train
Train model.
d4rl/rlkit/core/rl_algorithm.py:48
↓ 1 callersMethod_train_vae
(self, epoch)
d4rl/rlkit/torch/skewfit/online_vae_algorithm.py:74
↓ 1 callersFunction_type
The least generic type (type(None), int, float, str, unicode). >>> _type(None) is type(None) True >>> _type("foo") is type("") True
d4rl/rlkit/core/tabulate.py:244
↓ 1 callersMethod_update_info
(self, info, obs)
d4rl/rlkit/envs/vae_wrapper.py:119
↓ 1 callersMethod_update_target_networks
(self)
d4rl/rlkit/torch/ddpg/ddpg.py:178
↓ 1 callersMethod_validate_hyperparameters
(self)
d4rl/rlkit/util/hyperparameter.py:118
↓ 1 callersFunctionadd_border
(img, pad_length, pad_color, imsize=84)
d4rl/rlkit/torch/skewfit/video_gen.py:15
↓ 1 callersFunctionadd_border
(img, pad_length, pad_color, imsize=84)
d4rl/rlkit/util/video.py:108
↓ 1 callersMethodadd_decoded_vae_goals_to_path
(self, path)
d4rl/rlkit/data_management/online_vae_replay_buffer.py:95
↓ 1 callersMethodadd_path
Add a path to the replay buffer. This default implementation naively goes through every step, but you may want to optimize t
d4rl/rlkit/data_management/replay_buffer.py:33
↓ 1 callersMethodadd_sample
Add a transition tuple.
d4rl/rlkit/data_management/replay_buffer.py:10
↓ 1 callersMethodadd_text_output
(self, file_name)
d4rl/rlkit/core/logging.py:114
↓ 1 callersFunctionatanh
(x)
d4rl/rlkit/torch/sac/policies.py:15
← previousnext →101–200 of 729, ranked by callers