VisionData.infer_on_batch#

abstract VisionData.infer_on_batch(batch, model, device) → Union[List[Tensor], Tensor][source]#

Infer on a batch of data.

previous

VisionData.get_transform_type

next

VisionData.init_cache