Memories

[source]

Memory

polyaxon.rl.memories.Memory(size=1000, batch_size=32)

Base Agent Memory class.

  • Args:

    • size: int. The size of the memory.
    • batch_size: int. The batch size to return during the sampling.
  • Attributes:

    • _size: int. The size of the memory.
    • batchsize: int. The batch size to return during the sampling.
    • _memory: deque. Where to store the data.
    • _counter: int. Number of step stored up to size.
    • _spec: list. the list of keys corresponding to the data values stored each step.

[source]

BatchMemory

polyaxon.rl.memories.BatchMemory(batch_size=5000)

The batch memory buffer batch size data and clear the memory after each sample.