RuntimeError: expected input[100, 3, 512, 32] to have 131 channels

[pramishp]

I am trying to run the training on my Ubuntu 18.04.5 with python=3.7.10, torch=1.1.0. However, I am getting the following error when trying to run training with batchsize=100

......
  File "VoteHMR/src/models/backbone_module.py", line 95, in forward
    xyz, features, fps_inds = self.sa2(xyz, features)  # this fps_inds is just 0,1,...,1023
  File "/home/pramish/miniconda3/envs/votehmr/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "VoteHMR/src/pointnet2/pointnet2_modules.py", line 253, in forward
    grouped_features
  File "/home/pramish/miniconda3/envs/votehmr/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/pramish/miniconda3/envs/votehmr/lib/python3.7/site-packages/torch/nn/modules/container.py", line 92, in forward
    input = module(input)
  File "/home/pramish/miniconda3/envs/votehmr/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/pramish/miniconda3/envs/votehmr/lib/python3.7/site-packages/torch/nn/modules/container.py", line 92, in forward
    input = module(input)
  File "/home/pramish/miniconda3/envs/votehmr/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/pramish/miniconda3/envs/votehmr/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 338, in forward
    self.padding, self.dilation, self.groups)
RuntimeError: Given groups=1, weight of size 128 131 1 1, expected input[100, 3, 512, 32] to have 131 channels, but got 3 channels instead