MCPcopy Create free account

hub / github.com/danijar/dreamerv3 / functions

Functions705 in github.com/danijar/dreamerv3

⨍ Functions 705◇ Types & classes 110
Method__init__
( self, level, repeat=4, size=(64, 64), mode='train', actions='popart', episodic=True, text=None,
embodied/envs/dmlab.py:15
Method__init__
(self, obs_space, act_space, addr=None)
embodied/tests/utils.py:10
Method__init__
(self, env, when)
embodied/tests/test_driver.py:89
Method__init__
(self, modules, opt)
embodied/jax/opt.py:20
Method__init__
(self, impl)
embodied/jax/utils.py:24
Method__init__
(self, model, *, source, rate=1.0, every=1)
embodied/jax/utils.py:96
Method__init__
(self, module, count, names=('__call__',))
embodied/jax/utils.py:132
Method__init__
(self, dist='trunc_normal', fan='in', scale=1.0)
embodied/jax/nets.py:146
Method__init__
(self, classes, units, shape=())
embodied/jax/nets.py:205
Method__init__
(self, units)
embodied/jax/nets.py:237
Method__init__
(self, units, blocks)
embodied/jax/nets.py:261
Method__init__
(self, depth, kernel, stride=1)
embodied/jax/nets.py:294
Method__init__
(self, depth, kernel, stride=1)
embodied/jax/nets.py:335
Method__init__
(self, impl)
embodied/jax/nets.py:368
Method__init__
(self, spaces, fdims, squish=lambda x: x)
embodied/jax/nets.py:469
Method__init__
(self, spaces, units)
embodied/jax/nets.py:513
Method__init__
(self, layers=5, units=1024)
embodied/jax/nets.py:573
Method__init__
(self, output, dims, agg=jnp.sum)
embodied/jax/outs.py:42
Method__init__
(self, output)
embodied/jax/outs.py:81
Method__init__
(self, outputs, midpoints, axis)
embodied/jax/outs.py:101
Method__init__
(self, mean, squash=None)
embodied/jax/outs.py:131
Method__init__
(self, mean, eps=1.0)
embodied/jax/outs.py:146
Method__init__
(self, mean, stddev=1.0)
embodied/jax/outs.py:163
Method__init__
(self, logit)
embodied/jax/outs.py:191
Method__init__
(self, logits, unimix=0.0)
embodied/jax/outs.py:210
Method__init__
(self, logits, unimix=0.0)
embodied/jax/outs.py:245
Method__init__
(self, logits, bins, squash=None, unsquash=None)
embodied/jax/outs.py:275
Method__init__
(self, space, output, **hkw)
embodied/jax/heads.py:26
Method__init__
(self, spaces, outputs, **kw)
embodied/jax/heads.py:46
Method__init__
(self, space, output, **kw)
embodied/jax/heads.py:73
Method__init__
(self, obs_space, act_space, config)
dreamerv3/agent.py:33
Method__init__
(self, act_space, **kw)
dreamerv3/rssm.py:34
Method__init__
(self, obs_space, **kw)
dreamerv3/rssm.py:192
Method__init__
(self, obs_space, **kw)
dreamerv3/rssm.py:268
Method__iter__
(self)
embodied/core/base.py:63
Method__iter__
(self)
embodied/core/streams.py:19
Method__iter__
(self)
embodied/core/streams.py:44
Method__iter__
(self)
embodied/core/streams.py:116
Method__iter__
(self)
embodied/core/streams.py:161
Method__iter__
(self)
embodied/core/streams.py:188
Method__iter__
(self)
embodied/core/streams.py:217
Method__len__
(self)
embodied/core/selectors.py:15
Method__len__
(self)
embodied/core/selectors.py:37
Method__len__
(self)
embodied/core/selectors.py:71
Method__len__
(self)
embodied/core/selectors.py:160
Method__len__
(self)
embodied/core/selectors.py:241
Method__len__
(self)
embodied/core/selectors.py:324
Method__len__
(self)
embodied/core/replay.py:55
Method__len__
(self)
embodied/core/wrappers.py:13
Method__lt__
(self, other)
embodied/core/chunk.py:27
Method__new__
(subcls, obs_space, act_space, config)
embodied/jax/agent.py:38
Method__next__
(self)
embodied/core/base.py:66
Method__next__
(self)
embodied/core/streams.py:22
Method__next__
(self)
embodied/core/streams.py:50
Method__next__
(self)
embodied/core/streams.py:120
Method__next__
(self)
embodied/core/streams.py:167
Method__next__
(self)
embodied/core/streams.py:194
Method__next__
(self)
embodied/core/streams.py:221
Method__repr__
(self)
embodied/core/selectors.py:318
Method__repr__
(self)
embodied/core/base.py:36
Method__repr__
(self)
embodied/core/chunk.py:24
Method__repr__
(self)
embodied/jax/nets.py:180
Method__repr__
(self)
embodied/jax/outs.py:13
Method__repr__
(self)
embodied/jax/outs.py:47
Method__setitem__
(self, key, stepids)
embodied/core/selectors.py:18
Method__setitem__
(self, key, stepids)
embodied/core/selectors.py:45
Method__setitem__
(self, key, stepids)
embodied/core/selectors.py:89
Method__setitem__
(self, key, stepids)
embodied/core/selectors.py:170
Method__setitem__
(self, key, stepids)
embodied/core/selectors.py:217
Method_build
(self, color=[0.8, 0.8, 0.8], model='labmaze_style_01')
embodied/envs/loconav.py:101
Method_build
(self, name='walker', initializer=None)
embodied/envs/loconav_quadruped.py:17
Method_build_observables
(self)
embodied/envs/loconav_quadruped.py:33
Method_env_server
(stop, envid, pipe, ctor)
embodied/core/driver.py:103
Method_foot_bodies
(self)
embodied/envs/loconav_quadruped.py:65
Method_make_agent
(self, queue)
embodied/tests/test_parallel.py:58
Method_make_logger
(self)
embodied/tests/test_parallel.py:73
Method_make_logger
(self)
embodied/tests/test_train.py:49
Method_make_replay
(self, args, train_ratio=None)
embodied/tests/test_parallel.py:67
Method_make_replay
(self, args)
embodied/tests/test_train.py:45
Method_notempty
(self, reason=False)
embodied/core/replay.py:390
Method_scaled_winit
(self, *args, **kwargs)
embodied/jax/nets.py:250
Method_scaled_winit
(self, *args, **kwargs)
embodied/jax/nets.py:280
Method_scaled_winit
(self, *args, **kwargs)
embodied/jax/nets.py:322
Function_to_local
(x)
embodied/jax/internal.py:164
Method_watcher
(self)
embodied/tests/utils.py:23
Method_worker
(self)
embodied/core/streams.py:70
Method_wrapper
(self, fn, *args, **kwargs)
embodied/jax/outs.py:93
Method_wrapper
(self, fns, *args, **kwargs)
embodied/jax/outs.py:116
Methodact_space
(self)
embodied/core/base.py:50
Methodact_space
(self)
embodied/core/wrappers.py:100
Methodact_space
(self)
embodied/core/wrappers.py:216
Methodact_space
(self)
embodied/core/wrappers.py:282
Methodact_space
(self)
embodied/envs/atari.py:86
Methodact_space
(self)
embodied/envs/loconav.py:67
Methodact_space
(self)
embodied/envs/crafter.py:40
Methodact_space
(self)
embodied/envs/from_gym.py:48
Methodact_space
(self)
embodied/envs/dmc.py:60
Methodact_space
(self)
embodied/envs/dummy.py:31
Methodact_space
(self)
embodied/envs/pinpad.py:49
Methodact_space
(self)
embodied/envs/bsuite.py:32
← previousnext →301–400 of 705, ranked by callers