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/core/limiters.py:34

(cls, filename, error='raise')

embodied/core/chunk.py:77

embodied/tests/utils.py:102

(self, data, regex=None)

embodied/jax/agent.py:356

Functionload_run

(filename, xkeys, ykeys, ythres=None)

embodied/run/train.py:31

(tran, worker, mode)

embodied/run/train_eval.py:44

embodied/run/eval_only.py:27

embodied/jax/outs.py:174

embodied/jax/outs.py:197

embodied/jax/outs.py:226

embodied/jax/outs.py:256

embodied/tests/test_layer_scan.py:52

embodied/jax/outs.py:21

embodied/jax/outs.py:56

embodied/jax/outs.py:138

embodied/jax/outs.py:154

embodied/jax/outs.py:311

(self, carry, tokens, acts, reset, training)

dreamerv3/rssm.py:120

(*args, **kwargs)

embodied/jax/opt.py:34

(devices, shape, names)

embodied/jax/internal.py:226

Methodmjcf_model

embodied/envs/loconav_quadruped.py:37

Methodmocap_joints

embodied/envs/loconav_quadruped.py:61

Methodmocap_tracking_bodies

embodied/envs/loconav_quadruped.py:57

(xs, dst_sharding)

embodied/jax/internal.py:216

embodied/jax/heads.py:117

embodied/core/chunk.py:36

Methodnormal_logstd

embodied/jax/heads.py:157

Functionnormalize

Methodobs_space

embodied/core/base.py:43

Methodobs_space

embodied/core/wrappers.py:212

Methodobs_space

embodied/core/wrappers.py:305

Methodobs_space

embodied/core/wrappers.py:352

Methodobs_space

embodied/core/wrappers.py:375

Methodobs_space

embodied/envs/atari.py:76

Methodobs_space

embodied/envs/loconav.py:61

Methodobs_space

embodied/envs/crafter.py:24

Methodobs_space

embodied/envs/from_gym.py:33

Methodobs_space

embodied/envs/dmc.py:50

Methodobs_space

embodied/envs/dummy.py:16

Methodobs_space

embodied/envs/pinpad.py:56

Methodobs_space

embodied/envs/bsuite.py:28

Methodobs_space

embodied/envs/procgen.py:32

Methodobs_space

embodied/envs/minecraft_flat.py:224

Methodobs_space

embodied/envs/from_dm.py:22

Methodobs_space

embodied/envs/dmlab.py:55

Methodobservable_joints

embodied/envs/loconav_quadruped.py:78

embodied/jax/heads.py:112

Functionparallel_actor

(agent, barrier, args)

embodied/run/parallel.py:74

Functionparallel_agent

(make_agent, args)

embodied/run/parallel.py:62

Functionparallel_env

(make_env, envid, args, is_eval=False)

embodied/run/parallel.py:417

Functionparallel_envs

(make_env, make_env_eval, args)

embodied/run/parallel.py:475

Functionparallel_learner

(agent, barrier, args)

embodied/run/parallel.py:131

Functionparallel_logger

(make_logger, args)

embodied/run/parallel.py:318

Functionparallel_replay

(make_replay_train, make_replay_eval, make_stream, args)

embodied/run/parallel.py:221

(self, carry, obs, mode='train')

embodied/core/random.py:19

(self, carry, obs, mode='train')

embodied/tests/utils.py:44

(carry, obs, mode='train')

embodied/tests/test_driver.py:66

(self, carry, obs, mode='train')

embodied/jax/agent.py:220

(self, carry, obs, mode='train')

dreamerv3/agent.py:115

Methodpolicy_keys

dreamerv3/agent.py:86

embodied/run/parallel.py:111

embodied/jax/outs.py:135

embodied/jax/outs.py:151

embodied/jax/outs.py:167

embodied/jax/outs.py:194

embodied/jax/outs.py:219

embodied/jax/outs.py:248

embodied/jax/outs.py:285

Methodprev_action

embodied/envs/loconav_quadruped.py:96

embodied/jax/outs.py:30

Methodproprioception

embodied/envs/loconav_quadruped.py:124

(context, outer_addr, inner_addr, barrier)

embodied/perf/test_distr.py:84

embodied/core/selectors.py:337

(self, carry, data)

embodied/core/random.py:29

(self, carry, data)

embodied/jax/agent.py:316

(self, carry, data)

dreamerv3/agent.py:247

embodied/jax/nets.py:120

Methodroot_body

embodied/envs/loconav_quadruped.py:49

Methodroot_global_pos

embodied/envs/loconav_quadruped.py:109

embodied/envs/loconav_quadruped.py:110

embodied/core/limiters.py:78

(self, seed, shape=())

embodied/jax/outs.py:24

(self, seed, shape=())

embodied/jax/outs.py:170

(self, seed, shape=())

embodied/jax/outs.py:203

(self, seed, shape=())

embodied/jax/outs.py:222

(self, seed, shape=())

embodied/jax/outs.py:252

Functionsample_batch_eval

embodied/run/parallel.py:282

Functionsample_batch_report

embodied/run/parallel.py:278

Functionsample_batch_train

embodied/run/parallel.py:265

embodied/tests/test_replay.py:324

embodied/core/base.py:69

embodied/core/random.py:35

embodied/core/streams.py:25

embodied/core/streams.py:59

embodied/core/streams.py:142

embodied/core/streams.py:198

embodied/core/streams.py:228

embodied/core/limiters.py:31

(self, directory, log=False)

embodied/core/chunk.py:65

← previous next →501–600 of 705, ranked by callers