Skip to content

allenact.algorithms.onpolicy_sync.storage#

[view_source]

RolloutStorage#

class RolloutStorage(object)

[view_source]

Class for storing rollout information for RL trainers.

RolloutStorage.narrow#

 | narrow(num_steps=None)

[view_source]

This function is used by the training engine to temporarily (after one interrupted rollout in decentralized distributed settings, without arguments) or permanently (for a training stage with shorter horizon, with arguments) narrow the step dimension in the storage.

The reverse operation, unnarrow, is automatically called by after_update (without arguments) or when the rollout length varies in the training pipeline (with arguments).

RolloutStorage.unnarrow#

 | unnarrow(unnarrow_to_maximum_size=False)

[view_source]

See doc string for the narrow method.