adding docstring for abstractcacheddataloaderv2

This commit is contained in:
Jake Austin committed 2022-07-17 23:05:06 -07:00
1 parent 667de0b758
commit 0422afc913
1 file changed
+6 -1
+6 -1
View File
@@ -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