"ValueError: expected 5D input (got 4D input)" when generating point cloud

[aidan-sonoferick]

Hello,

I am trying to generate point clouds for scenes 1 and 3. However, when I went to run generate_point_clouds in network_time.py, I got this error on line 71 from the BatchNorm3D layer in DopplerEncoder. I have not modified any code outside of the params (to change the test scenes to [1,3]), and the checkpoint path.

...
File "/afs/ece.cmu.edu/usr/aerickso/RaDelft-Dataset/machine_learning_python/networks/network_time.py", line 72, in forward
    x = self.norm1(x)
        ^^^^^^^^^^^^^
  File "/afs/ece.cmu.edu/usr/aerickso/miniconda3/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/afs/ece.cmu.edu/usr/aerickso/miniconda3/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/afs/ece.cmu.edu/usr/aerickso/miniconda3/lib/python3.12/site-packages/torch/nn/modules/batchnorm.py", line 142, in forward
    self._check_input_dim(input)
  File "/afs/ece.cmu.edu/usr/aerickso/miniconda3/lib/python3.12/site-packages/torch/nn/modules/batchnorm.py", line 536, in _check_input_dim
    raise ValueError(f"expected 5D input (got {input.dim()}D input)")
ValueError: expected 5D input (got 4D input)
-----------------------------------------

Is there something wrong in the function? Thanks for any help you can provide :)