Functions — github.com/bytetriper/RAE

CodeHub MCPcopy Create free account

hub / github.com/bytetriper/RAE / functions

Functions315 in github.com/bytetriper/RAE

⨍ Functions 315◇ Types & classes 71

Functionlr_lambda

src/utils/optim_utils.py:89

Methodnoise_ode

(x, t, model, **model_kwargs)

src/stage2/transport/transport.py:239

Functionnone_or_str

src/utils/train_utils.py:31

Methodprior_logp

Standard multivariate normal prior Assume z is batched

src/stage2/transport/transport.py:114

Functionrandom_sampler

(_step_idx: int)

src/sample_ddp.py:65

Functionregister_encoder

Register an encoder class in ``ARCHS``. Can be used either as ``@register_encoder()`` (optionally passing ``name``) or via ``register_encoder

src/stage1/encoders/__init__.py:16

Functionrequires_grad

Set requires_grad flag for all parameters in a model.

src/utils/train_utils.py:60

forward loop of sde

src/stage2/transport/integrators.py:69

(self, x, model, **model_kwargs)

src/stage2/transport/integrators.py:106

Methodscore_ode

(x, t, model, **model_kwargs)

src/stage2/transport/transport.py:234

Methodsde_diffusion_fn

src/stage2/transport/transport.py:304

Methodsde_drift_fn

(x, t, model, **kwargs)

src/stage2/transport/transport.py:308

Functionvanilla_d_loss

Original GAN discriminator loss.

src/disc/gan_loss.py:15

Functionvanilla_g_loss

Original GAN generator loss.

src/disc/gan_loss.py:23

Methodvelocity_ode

(x, t, model, **model_kwargs)

src/stage2/transport/transport.py:246

← previous301–315 of 315, ranked by callers