Add kwargs handling for patchpixelsampler patch size

[chungmin99]

PatchPixelSamplerConfig has a patch_size item, but the current PatchPixelSampler init function doesn't override the value of patch_size with the corresponding kwargs value.

This is particularly problematic, because PatchPixelSamplerConfig is generated within the datamanager:

nerfstudio/nerfstudio/data/datamanagers/base_datamanager.py

Line 472 in 9528a3f

return PatchPixelSamplerConfig().setup(

This PR simply adds the __init__ function for the sampler and adds a self.config.patch_size = self.kwargs.get(....

As an aside, it would be good if this override could happen automatically...

[brentyi]

Possibly dumb question: in the linked snippet, why are the kwargs used/needed at all?

Like instead of:

            return PatchPixelSamplerConfig().setup(
                patch_size=self.config.patch_size, num_rays_per_batch=num_rays_per_batch
            )

Would this work?

            return PatchPixelSamplerConfig(
                patch_size=self.config.patch_size,
                num_rays_per_batch=num_rays_per_batch,
            ).setup()

[chungmin99]

I believe the latter should work.

On another note (maybe address in this PR or in some future), perhaps it would be better to have people manually change the datamanager's pixel_sampler to PatchPixelSampler in the method configs, instead of automatically generating it under the hood when patch_size > 1.

nerfstudio/nerfstudio/data/datamanagers/base_datamanager.py

Line 348 in 9528a3f

pixel_sampler: PixelSamplerConfig = PixelSamplerConfig()