towhee.data.dataset.dataset.TorchDataSet

class towhee.data.dataset.dataset.TorchDataSet(torchDataSet: Optional[Dataset] = None)[source]

Bases: TowheeDataSet

pytorch dataset

Methods

get_framework

__init__(torchDataSet: Optional[Dataset] = None)[source]