LitDaliWrapper

class lightning_pose.data.dali.LitDaliWrapper(*args, eval_mode: Literal['train', 'predict'], num_iters: int = 1, do_context: bool = False, **kwargs)[source]

Bases: DALIGenericIterator

wrapper around a DALI pipeline to get batches for ptl.

__init__(*args, eval_mode: Literal['train', 'predict'], num_iters: int = 1, do_context: bool = False, **kwargs) → None[source]

Wrapper around DALIGenericIterator to get batches for pl.

Parameters:

  • eval_mode

  • num_iters – number of enumerations of dataloader (should be computed outside for now; should be fixed by lightning/dali teams)

  • do_context – whether model/loader use 5-frame context or not