ertk.pytorch.dataset.MTLDataset

class ertk.pytorch.dataset.MTLDataset(x: Tensor, y: dict[str, torch.Tensor])

Bases: Dataset[tuple[Tensor, dict[str, Tensor]]]

__init__(x: Tensor, y: dict[str, torch.Tensor])

Methods

__init__(x, y)

Inherited Methods