ray.rllib.policy.Policy.load_batch_into_buffer
ray.rllib.policy.Policy.load_batch_into_buffer#
- Policy.load_batch_into_buffer(batch: ray.rllib.policy.sample_batch.SampleBatch, buffer_index: int = 0) int [source]#
Bulk-loads the given SampleBatch into the devices’ memories.
The data is split equally across all the Policy’s devices. If the data is not evenly divisible by the batch size, excess data should be discarded.
- Parameters
batch – The SampleBatch to load.
buffer_index – The index of the buffer (a MultiGPUTowerStack) to use on the devices. The number of buffers on each device depends on the value of the
num_multi_gpu_tower_stacks
config key.
- Returns
The number of tuples loaded per device.