module emote.memory.builder

Classes

class DictMemoryTable(ArrayMemoryTable):

Methods

def __init__(
    self
,
    *use_terminal_column,
    obs_keys,
    columns,
    maxlen,
    length_key,
    sampler,
    device
) -> None

class DictObsMemoryTable(DictMemoryTable):

Create a memory suited for Reinforcement Learning Tasks with 1-Step Bellman Backup with a single bootstrap value, and using dictionary observations as network inputs.

Methods

def __init__(
    self
,
    *spaces,
    use_terminal_column,
    maxlen,
    device,
    dones_dtype,
    masks_dtype,
    sampler
) -> None

Arguments:

  • spaces(MDPSpace)
  • use_terminal_column(bool)
  • maxlen(int)
  • device(torch.device)
  • dones_dtype
  • masks_dtype
  • sampler(SampleStrategy)

class DictObsNStepMemoryTable(DictMemoryTable):

Create a memory suited for Reinforcement Learning Tasks with N-Step Bellman Backup with a single bootstrap value, and using dictionary observations as network inputs.

Methods

def __init__(self, *spaces, use_terminal_column, maxlen, sampler, device) -> None

Arguments:

  • spaces(MDPSpace)
  • use_terminal_column(bool)
  • maxlen(int)
  • sampler(SampleStrategy)
  • device(torch.device)