module emote.algorithms.genrl.vae

Classes

class VariationalAutoencoder(nn.Module):

Methods

def __init__(self, encoder, decoder, device, beta) -> None

def forward(self, x, condition) -> None

def loss(self, x, x_hat, mu, log_std) -> None

class VAELoss(LossCallback):

Methods

def __init__(
    self
,
    *vae,
    opt,
    lr_schedule,
    max_grad_norm,
    name,
    data_group,
    input_key,
    conditioning_func
) -> None

def loss(self, observation, actions) -> None