nip.scenario_base.data.TensorDictSingleProcessDataLoaderIter

nip.scenario_base.data.TensorDictSingleProcessDataLoaderIter#

class nip.scenario_base.data.TensorDictSingleProcessDataLoaderIter(loader)[source]#

Single process DataLoaderIter for tensordicts.

This is a hack to allow the DataLoader to work with TensorDict memory pinning.

Methods Summary

Methods

_next_data()[source]#