MCPcopy Create free account

hub / github.com/Howuhh/prioritized_experience_replay / functions

Functions26 in github.com/Howuhh/prioritized_experience_replay

↓ 14 callersFunctiondevice
(force_cpu=True)
memory/utils.py:15
↓ 3 callersMethodget
(self, cumsum)
memory/tree.py:43
↓ 3 callersMethodsample
(self, batch_size)
memory/buffer.py:139
↓ 3 callersMethodupdate
(self, batch, weights=None)
train.py:38
↓ 2 callersMethodact
(self, state)
train.py:32
↓ 2 callersMethodadd
(self, value, data)
memory/tree.py:36
↓ 2 callersFunctionrun_experiment
(config, use_priority=False, n_seeds=10)
train.py:147
↓ 2 callersMethodupdate
(self, data_idx, value)
memory/tree.py:25
↓ 1 callersMethodadd
(self, transition)
memory/buffer.py:125
↓ 1 callersFunctionevaluate_policy
(env_name, agent, episodes=5, seed=0)
train.py:66
↓ 1 callersFunctionmain
()
plots/sampling.py:34
↓ 1 callersFunctionsample_cumsum
(tree, batch_size)
plots/sampling.py:23
↓ 1 callersFunctionsample_segments
(tree, batch_size)
plots/sampling.py:8
↓ 1 callersMethodsave
(self)
train.py:62
↓ 1 callersFunctionset_seed
(env, seed=0)
memory/utils.py:7
↓ 1 callersMethodsoft_update
(self, target, source)
train.py:28
↓ 1 callersFunctiontrain
(env_name, model, buffer, timesteps=200_000, batch_size=128, eps_max=0.1, eps_min=0.0, test_every=50
train.py:83
↓ 1 callersMethodupdate_priorities
(self, data_idxs, priorities)
memory/buffer.py:98
Method__init__
(self, state_size, action_size, gamma, tau, lr)
train.py:14
Method__init__
(self, size)
memory/tree.py:13
Method__init__
(self, state_size, action_size, buffer_size, eps=1e-2, alpha=0.1, beta=0.1)
memory/buffer.py:10
Method__init__
(self, state_size, action_size, buffer_size)
memory/buffer.py:113
Method__repr__
(self)
memory/tree.py:60
Methodadd
(self, transition)
memory/buffer.py:30
Methodsample
(self, batch_size)
memory/buffer.py:47
Methodtotal
(self)
memory/tree.py:22