From 0422afc9135ae0f61218d24b998354f91d4e8cc1 Mon Sep 17 00:00:00 2001 From: Jake Austin Date: Sun, 17 Jul 2022 23:05:06 -0700 Subject: [PATCH] adding docstring for abstractcacheddataloaderv2 --- pyrad/data/dataloader.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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