Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/TianhongDai/reinforcement-learning-algorithms
/ functions
Functions
284 in github.com/TianhongDai/reinforcement-learning-algorithms
⨍
Functions
284
◇
Types & classes
57
↓ 25 callers
Method
mean
(self)
rl_algorithms/dqn_algos/utils.py:48
↓ 25 callers
Method
reset
(self)
rl_algorithms/ddpg/utils.py:15
↓ 25 callers
Method
step
(self, action)
rl_utils/logger/bench.py:51
↓ 16 callers
Method
close
(self)
rl_utils/logger/logger.py:352
↓ 10 callers
Method
sample
Gradients will and should *not* pass through this operation. See https://github.com/pytorch/pytorch/issues/4620 for discussion.
rl_algorithms/sac/utils.py:35
↓ 9 callers
Function
get_current
()
rl_utils/logger/logger.py:289
↓ 9 callers
Method
step
(self, ac)
rl_utils/env_wrapper/atari_wrapper.py:301
↓ 8 callers
Method
__init__
Warp frames to 84x84 as done in the Nature paper and later work. If the environment uses dictionary observations, `dict_space_key` c
rl_utils/env_wrapper/atari_wrapper.py:138
↓ 8 callers
Function
logkv
Log a value of some diagnostic Call this once for each diagnostic quantity, each iteration If called many times, last value will be used.
rl_utils/logger/logger.py:193
↓ 8 callers
Method
render
(self, mode='human')
rl_utils/env_wrapper/__init__.py:108
↓ 8 callers
Method
reset
(self, **kwargs)
rl_utils/env_wrapper/atari_wrapper.py:309
↓ 6 callers
Method
close
(self)
rl_utils/env_wrapper/__init__.py:91
↓ 6 callers
Function
log
Write the sequence of args, with no separators, to the console and output files (if you've configured an output file).
rl_utils/logger/logger.py:224
↓ 6 callers
Method
log_prob
:param value: some value, x :param pre_tanh_value: arctanh(x) :return:
rl_algorithms/sac/utils.py:25
↓ 5 callers
Method
_force
(self)
rl_utils/env_wrapper/atari_wrapper.py:244
↓ 5 callers
Function
create_single_env
(args, rank=0)
rl_utils/env_wrapper/create_env.py:13
↓ 5 callers
Function
dumpkvs
Write all of the diagnostics from the current iteration
rl_utils/logger/logger.py:214
↓ 5 callers
Method
get_dir
(self)
rl_utils/logger/logger.py:349
↓ 5 callers
Function
make_atari
(env_id, max_episode_steps=None)
rl_utils/env_wrapper/atari_wrapper.py:269
↓ 5 callers
Method
select_actions
(self, exploration=True, reparameterize=True)
rl_algorithms/sac/utils.py:67
↓ 5 callers
Function
set_seeds
(args, rank=0)
rl_utils/seeds/seeds.py:6
↓ 5 callers
Function
wrap_deepmind
Configure environment for DeepMind-style Atari.
rl_utils/env_wrapper/atari_wrapper.py:278
↓ 4 callers
Method
_assert_not_closed
(self)
rl_utils/env_wrapper/multi_envs_wrapper.py:100
↓ 4 callers
Method
add
(self, obs, action, reward, obs_, done)
rl_utils/experience_replay/experience_replay.py:16
↓ 4 callers
Method
log
(self, *args, level=INFO)
rl_utils/logger/logger.py:337
↓ 3 callers
Method
__init__
(self, num_envs, observation_space, action_space)
rl_utils/env_wrapper/__init__.py:41
↓ 3 callers
Method
_get_tensor_inputs
(self, obs)
rl_algorithms/sac/sac_agent.py:134
↓ 3 callers
Method
_get_tensors
(self, obs)
rl_algorithms/a2c/a2c_agent.py:126
↓ 3 callers
Method
_get_tensors
(self, obs)
rl_algorithms/ppo/ppo_agent.py:200
↓ 3 callers
Method
_get_tensors
(self, obs)
rl_algorithms/dqn_algos/dqn_agent.py:116
↓ 3 callers
Method
_mpi_average
(self, x)
rl_utils/mpi_utils/normalizer.py:60
↓ 3 callers
Function
configure
If comm is provided, average all numerical stats across that comm
rl_utils/logger/logger.py:372
↓ 3 callers
Function
info
(*args)
rl_utils/logger/logger.py:233
↓ 3 callers
Method
normalize
(self, v, clip_range=None)
rl_utils/mpi_utils/normalizer.py:67
↓ 3 callers
Function
plot_results
(dirs, num_timesteps, xaxis, task_name, plt_order, beta=False)
rl_utils/logger/plot.py:67
↓ 3 callers
Method
update
(self, ob, rew, done, info)
rl_utils/logger/bench.py:58
↓ 2 callers
Method
__init__
(self, state_size, num_actions, dist_type)
rl_algorithms/ppo/models.py:11
↓ 2 callers
Method
__init__
(self, num_states, num_actions)
rl_algorithms/trpo/models.py:31
↓ 2 callers
Function
_flatten_obs
(obs)
rl_utils/env_wrapper/multi_envs_wrapper.py:107
↓ 2 callers
Function
_get_flat_params_or_grads
include two kinds: grads and params
rl_utils/mpi_utils/utils.py:25
↓ 2 callers
Method
_get_ob
(self)
rl_utils/env_wrapper/atari_wrapper.py:218
↓ 2 callers
Method
_get_tensors
(self, obs)
rl_algorithms/trpo/trpo_agent.py:173
↓ 2 callers
Method
_preproc_inputs
(self, obs)
rl_algorithms/ddpg/ddpg_agent.py:110
↓ 2 callers
Function
_set_flat_params_or_grads
include two kinds: grads and params
rl_utils/mpi_utils/utils.py:33
↓ 2 callers
Method
_soft_update_target_network
(self, target, source)
rl_algorithms/ddpg/ddpg_agent.py:163
↓ 2 callers
Method
_truncate
(self, s)
rl_utils/logger/logger.py:71
↓ 2 callers
Method
_update_target_network
(self, target, source)
rl_algorithms/sac/sac_agent.py:190
↓ 2 callers
Function
create_multiple_envs
(args)
rl_utils/env_wrapper/create_env.py:37
↓ 2 callers
Function
debug
(*args)
rl_utils/logger/logger.py:230
↓ 2 callers
Function
denormalize
(x, mean, std, clip=10)
rl_algorithms/trpo/demo.py:7
↓ 2 callers
Function
discount_with_dones
(rewards, dones, gamma)
rl_algorithms/a2c/utils.py:18
↓ 2 callers
Function
eval_actions
(pi, actions)
rl_algorithms/trpo/utils.py:12
↓ 2 callers
Function
evaluate_actions
(pi, actions, dist_type, env_type)
rl_algorithms/ppo/utils.py:21
↓ 2 callers
Function
get_args
()
rl_algorithms/sac/arguments.py:4
↓ 2 callers
Function
get_args
()
rl_algorithms/ddpg/arguments.py:3
↓ 2 callers
Function
get_args
()
rl_algorithms/a2c/arguments.py:3
↓ 2 callers
Function
get_args
()
rl_algorithms/ppo/arguments.py:3
↓ 2 callers
Function
get_args
()
rl_algorithms/dqn_algos/arguments.py:3
↓ 2 callers
Function
get_args
()
rl_algorithms/trpo/arguments.py:3
↓ 2 callers
Method
get_log_prob
(self, actions, pre_tanh_value)
rl_algorithms/sac/utils.py:78
↓ 2 callers
Function
logkv_mean
The same as logkv(), but if called many times, values averaged.
rl_utils/logger/logger.py:201
↓ 2 callers
Method
process
(self, obs)
rl_utils/env_wrapper/__init__.py:169
↓ 2 callers
Method
reset
(self)
rl_utils/env_wrapper/multi_envs_wrapper.py:77
↓ 2 callers
Function
select_actions
(pi, deterministic=False)
rl_algorithms/a2c/utils.py:6
↓ 2 callers
Function
set_flat_params_to
(model, flat_params)
rl_algorithms/trpo/utils.py:53
↓ 2 callers
Method
std
(self)
rl_utils/running_filter/running_filter.py:38
↓ 2 callers
Method
step_async
Tell all the environments to start taking a step with the given actions. Call step_wait() to get the results of the step.
rl_utils/env_wrapper/__init__.py:59
↓ 2 callers
Method
step_wait
Wait for the step taken with step_async(). Returns (obs, rews, dones, infos): - obs: an array of observations, or a dict of
rl_utils/env_wrapper/__init__.py:71
↓ 2 callers
Function
sync_grads
(network)
rl_utils/mpi_utils/utils.py:17
↓ 2 callers
Function
sync_networks
netowrk is the network you want to sync
rl_utils/mpi_utils/utils.py:6
↓ 1 callers
Method
__init__
(self, input_dims, hidden_size, action_dims=None)
rl_algorithms/sac/models.py:8
↓ 1 callers
Method
__init__
(self, obs_dims, action_dims)
rl_algorithms/ddpg/models.py:7
↓ 1 callers
Method
__init__
(self, num_actions)
rl_algorithms/a2c/models.py:34
↓ 1 callers
Method
__init__
(self, num_actions, use_dueling=False)
rl_algorithms/dqn_algos/models.py:32
↓ 1 callers
Method
_adjust_learning_rate
(self, update, num_updates)
rl_algorithms/ppo/ppo_agent.py:211
↓ 1 callers
Function
_configure_default_logger
()
rl_utils/logger/logger.py:401
↓ 1 callers
Function
_demo
()
rl_utils/logger/logger.py:423
↓ 1 callers
Method
_do_log
(self, args)
rl_utils/logger/logger.py:358
↓ 1 callers
Method
_encode_sample
(self, idx)
rl_utils/experience_replay/experience_replay.py:26
↓ 1 callers
Method
_eval_agent
(self)
rl_algorithms/ddpg/ddpg_agent.py:168
↓ 1 callers
Method
_evaluate_agent
(self)
rl_algorithms/sac/sac_agent.py:195
↓ 1 callers
Method
_fisher_vector_product
(self, v, obs, pi_old)
rl_algorithms/trpo/trpo_agent.py:151
↓ 1 callers
Method
_get_kl
(self, obs, pi_old)
rl_algorithms/trpo/trpo_agent.py:164
↓ 1 callers
Method
_get_surrogate_loss
(self, obs, adv, actions, pi_old)
rl_algorithms/trpo/trpo_agent.py:143
↓ 1 callers
Method
_initial_exploration
(self, exploration_policy='gaussian')
rl_algorithms/sac/sac_agent.py:110
↓ 1 callers
Method
_select_actions
(self, pi)
rl_algorithms/ddpg/ddpg_agent.py:116
↓ 1 callers
Method
_update_network
(self)
rl_algorithms/ddpg/ddpg_agent.py:127
↓ 1 callers
Method
_update_network
(self, obs, returns, actions)
rl_algorithms/a2c/a2c_agent.py:98
↓ 1 callers
Method
_update_network
(self, obs, actions, returns, advantages)
rl_algorithms/ppo/ppo_agent.py:148
↓ 1 callers
Method
_update_network
(self, samples)
rl_algorithms/dqn_algos/dqn_agent.py:79
↓ 1 callers
Method
_update_network
(self, mb_obs, mb_actions, mb_returns, mb_advs)
rl_algorithms/trpo/trpo_agent.py:95
↓ 1 callers
Method
_update_newtork
(self)
rl_algorithms/sac/sac_agent.py:139
↓ 1 callers
Method
add_rewards
(self, reward)
rl_algorithms/dqn_algos/utils.py:31
↓ 1 callers
Function
clear_mpi_env_vars
from mpi4py import MPI will call MPI_Init by default. If the child process has MPI environment variables, MPI will think that the child process
rl_utils/env_wrapper/__init__.py:197
↓ 1 callers
Method
close_extras
Clean up the extra resources, beyond what's in this base class. Only runs when not self.closed.
rl_utils/env_wrapper/__init__.py:84
↓ 1 callers
Function
conjugated_gradient
(fvp, b, update_steps, obs, pi_old, residual_tol=1e-10)
rl_algorithms/trpo/utils.py:18
↓ 1 callers
Method
dumpkvs
(self)
rl_utils/logger/logger.py:319
↓ 1 callers
Function
evaluate_actions
(pi, actions)
rl_algorithms/a2c/utils.py:14
↓ 1 callers
Method
get_images
Return RGB images from each environment
rl_utils/env_wrapper/__init__.py:111
↓ 1 callers
Function
get_rank_without_mpi_import
()
rl_utils/logger/logger.py:363
next →
1–100 of 284, ranked by callers