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.