module emote.nn.action_value_mlp
Classes
class ActionValueMlp(nn.Module):
Methods
def __init__(self, observation_dim, action_dim, hidden_dims) -> None
def forward(self, action, obs) -> Tensor
class SharedEncoderActionValueNet(nn.Module):
Methods
def __init__(self, shared_enc, encoder_out_dim, action_dim, hidden_dims) -> None
def forward(self, action, obs) -> None