docstring

This commit is contained in:
AdamRashid96 committed 2023-06-13 00:18:31 +00:00
1 parent 0fe3f539d6
commit f7f038fb41
1 file changed
+1
+1
View File
@@ -304,6 +304,7 @@ class PatchPixelSampler(PixelSampler):
class PairPixelSampler(PixelSampler): # pylint: disable=too-few-public-methods
"""Samples pair of pixels from 'image_batch's. Samples pairs of pixels from
from the images randomly within a 'radius' distance apart. Useful for pair-based losses.
Args:
num_rays_per_batch: number of rays to sample per batch
keep_full_image: whether or not to include a reference to the full image in returned batch