Results 3 issues of Bjoern

Could it be that [this](https://github.com/soravux/pms/blob/c85f431c71f1d1b485904f2b172d009723a527d8/pms.py#L93) should actually be changed from `Q[:, 4:] *= 2` to `Q[4:, :] *= 2`? As we don't want to multiply by two in the pixel...

I've a machine with two GPUs: ```bash $ nvidia-smi Thu Sep 7 13:43:12 2023 +---------------------------------------------------------------------------------------+ | NVIDIA-SMI 535.54.03 Driver Version: 535.54.03 CUDA Version: 12.2 | |-----------------------------------------+----------------------+----------------------+ | GPU Name Persistence-M...

I guess in [this line](https://github.com/krrish94/nerf-pytorch/blob/a14357da6cada433d28bf11a45c7bcaace76c06e/nerf/models.py#L243), `linear_layers` should be `layers_xyz`? In a similar realm, I guess in [this line](https://github.com/krrish94/nerf-pytorch/blob/a14357da6cada433d28bf11a45c7bcaace76c06e/nerf/models.py#L210) you don't want a skip connection in the last layer? Then `i...