tf_agents.replay_buffers.py_hashed_replay_buffer.FrameBuffer

View source on GitHub

Saves some frames in a memory efficient way.

Thread safety: cannot add multiple frames in parallel.

Methods

add_frame

View source

Add a frame to the buffer.

Args
frame Numpy array.

Returns
A deduplicated frame.

clear

View source

compress

View source

decompress

View source

deserialize

View source

Callback for PythonStateWrapper to deserialize the array.

on_delete

View source

serialize

View source

Callback for PythonStateWrapper to serialize the dictionary.

__len__

View source