Functions — github.com/danijar/dreamerv3

CodeHub MCPcopy Create free account

hub / github.com/danijar/dreamerv3 / functions

Functions705 in github.com/danijar/dreamerv3

⨍ Functions 705◇ Types & classes 110

embodied/tests/utils.py:98

embodied/jax/agent.py:340

embodied/perf/test_replay.py:111

Functionscale_by_momentum

(beta=0.9, nesterov=False)

embodied/jax/opt.py:146

Functionscale_by_rms

(beta=0.999, eps=1e-8)

embodied/jax/opt.py:126

(context, addr)

embodied/perf/test_bandwidth.py:35

(is_server, replica, replicas, port, addr)

embodied/core/clock.py:11

( platform=None, compute_dtype=jnp.bfloat16, debug=False, jit=True, prealloc=False, mo

embodied/jax/internal.py:15

embodied/core/wrappers.py:37

embodied/core/wrappers.py:63

embodied/core/wrappers.py:84

embodied/core/wrappers.py:106

embodied/core/wrappers.py:219

embodied/core/wrappers.py:251

embodied/core/wrappers.py:286

embodied/core/wrappers.py:312

embodied/core/wrappers.py:358

embodied/core/wrappers.py:402

embodied/envs/atari.py:92

embodied/envs/loconav.py:70

embodied/envs/crafter.py:46

embodied/envs/from_gym.py:57

embodied/envs/dummy.py:38

embodied/envs/pinpad.py:65

embodied/envs/bsuite.py:35

embodied/envs/procgen.py:41

embodied/envs/minecraft_flat.py:50

embodied/envs/minecraft_flat.py:101

embodied/envs/minecraft_flat.py:248

embodied/envs/from_dm.py:51

embodied/envs/dmlab.py:75

embodied/tests/test_driver.py:93

embodied/core/random.py:32

embodied/jax/agent.py:326

embodied/jax/nets.py:63

Methodsymexp_twohot

embodied/jax/heads.py:132

embodied/jax/nets.py:59

Methodsymlog_mse

embodied/jax/heads.py:127

Methodtest_agent_inputs

embodied/tests/test_driver.py:60

Methodtest_apply

(self, L=4, B=2, D=8)

embodied/tests/test_layer_scan.py:76

Methodtest_capacity_exact

(self, Replay, length, workers, capacity)

embodied/tests/test_replay.py:50

Methodtest_chunk_size

(self, chunksize, inserts=2e5, workers=8, samples=2e5)

embodied/perf/test_replay.py:50

Methodtest_depth_inserts

(self, inserts, branching)

embodied/tests/test_sampletree.py:20

Methodtest_depth_removals

(self, inserts, remove_every, branching)

embodied/tests/test_sampletree.py:32

Methodtest_env_reset

embodied/tests/test_driver.py:44

Methodtest_episode_length

embodied/tests/test_driver.py:9

Methodtest_first_step

embodied/tests/test_driver.py:18

Methodtest_grad

(self, L=4, B=2, D=8)

embodied/tests/test_layer_scan.py:88

Methodtest_infinity_probs

(self, branching)

embodied/tests/test_sampletree.py:162

Methodtest_init

(self, L=4, B=2, D=8)

embodied/tests/test_layer_scan.py:58

Methodtest_last_step

embodied/tests/test_driver.py:31

Methodtest_multiple_keys

embodied/tests/test_replay.py:36

Methodtest_parallel

(self, tmpdir, name, Replay, duration=5)

embodied/perf/test_replay.py:79

Methodtest_removal

(self, name, Replay, inserts=1e6, workers=1)

embodied/perf/test_replay.py:69

Methodtest_removal_num_nodes

(self, inserts, branching)

embodied/tests/test_sampletree.py:46

Methodtest_restore_chunks_exact

( self, tmpdir, Replay, length, capacity, chunksize)

embodied/tests/test_replay.py:218

Methodtest_restore_chunks_workers

( self, tmpdir, Replay, workers, length, capacity, chunksize)

embodied/tests/test_replay.py:252

Methodtest_restore_exact

(self, tmpdir, Replay, length, capacity, chunksize)

embodied/tests/test_replay.py:156

Methodtest_restore_insert

(self, tmpdir, Replay, length, capacity, chunksize)

embodied/tests/test_replay.py:284

Methodtest_restore_noclear

(self, tmpdir, Replay, length, capacity, chunksize)

embodied/tests/test_replay.py:177

Methodtest_restore_workers

(self, tmpdir, Replay, workers, length, capacity)

embodied/tests/test_replay.py:198

Methodtest_root_sum

(self, branching)

embodied/tests/test_sampletree.py:11

Methodtest_run_loop

(self, tmpdir, train_ratio, eval_envs)

embodied/tests/test_parallel.py:19

Methodtest_run_loop

embodied/tests/test_train.py:12

Methodtest_sample_frequencies

(self, scale, branching)

embodied/tests/test_sampletree.py:95

Methodtest_sample_sequences

( self, Replay, length, workers, capacity, chunksize)

embodied/tests/test_replay.py:63

Methodtest_sample_single

(self, Replay, length, capacity)

embodied/tests/test_replay.py:78

Methodtest_sample_single

(self, branching)

embodied/tests/test_sampletree.py:61

Methodtest_sample_uniform

embodied/tests/test_replay.py:88

Methodtest_sample_uniform

(self, inserts, branching, uprob)

embodied/tests/test_sampletree.py:74

Methodtest_speed

(self, name, Replay, inserts=2e5, workers=8, samples=1e5)

embodied/perf/test_replay.py:31

Methodtest_threading

( self, tmpdir, Replay, length=5, capacity=128, chunksize=32, adders=8, samplers=4)

embodied/tests/test_replay.py:307

Methodtest_throughput_crafter

(self, parallel=True)

embodied/perf/test_driver.py:27

Methodtest_throughput_dummy

(self, parallel=True)

embodied/perf/test_driver.py:13

Methodtest_unexpected_reset

embodied/tests/test_driver.py:85

Methodtest_update_frequencies

(self, branching)

embodied/tests/test_sampletree.py:117

Methodtest_worker_delay

(self, Replay, length, workers, capacity)

embodied/tests/test_replay.py:141

Methodtest_workers_random

(self, Replay, length=4, capacity=30)

embodied/tests/test_replay.py:117

Methodtest_workers_simple

embodied/tests/test_replay.py:105

Methodtest_zero_probs_mixed

(self, branching)

embodied/tests/test_sampletree.py:141

Methodtest_zero_probs_only

(self, branching)

embodied/tests/test_sampletree.py:154

Methodto_string

embodied/envs/minecraft_flat.py:434

Methodtorso_global_pos

embodied/envs/loconav_quadruped.py:116

Methodtorso_pos

embodied/envs/loconav_quadruped.py:117

(make_agent, make_replay, make_env, make_stream, make_logger, args)

embodied/run/train.py:9

(self, carry, data)

embodied/core/random.py:26

(self, carry, data)

embodied/jax/agent.py:264

Functiontrain_eval

( make_agent, make_replay_train, make_replay_eval, make_env_train, make_env_eval, make

embodied/run/train_eval.py:9

Functiontrainfn

embodied/run/train.py:70

Functiontrainfn

embodied/run/train_eval.py:93

embodied/run/parallel.py:344

(self, entries, carry=None)

dreamerv3/rssm.py:207

(self, entries, carry=None)

dreamerv3/rssm.py:285

embodied/jax/utils.py:113

Functionupdate_fn

(updates, state, params=None)

embodied/jax/opt.py:114

Methodupright_pose

embodied/envs/loconav_quadruped.py:41

(predicate, message, info=None, sleep=0.01, notify=60)

embodied/core/limiters.py:5

Methodwant_insert

embodied/core/limiters.py:38

Methodwant_sample

embodied/core/limiters.py:53

embodied/run/parallel.py:89

← previous next →601–700 of 705, ranked by callers