hoops_ai.dataset.TorchAdapter

class hoops_ai.dataset.TorchAdapter(cad_dataset, collate_fn=None)

Bases: object

Adapts CADDataset to PyTorch format

get_dataset()

Return a PyTorch Dataset wrapper