diff --git a/pyrad/data/dataloader.py b/pyrad/data/dataloader.py index 1b6489b10..186b51eae 100644 --- a/pyrad/data/dataloader.py +++ b/pyrad/data/dataloader.py @@ -193,7 +193,12 @@ class AbstractDataloaderV2(nn.Module): class AbstractCachedDataloaderV2(AbstractDataloaderV2): - """Subclass of the new V2 dataloader that is used for when things fit in memory""" + """Subclass of the new V2 dataloader that is used for when things fit in memory + + Attributes: + camera_to_world (torch.Tensor): camera to world transformation + intrinsics (torch.Tensor): intrinsics matrix + """ camera_to_world: torch.Tensor intrinsics: torch.Tensor