For custom point cloud data, classification fails during training.

[harishkashyap]

Traceback (most recent call last):
File "train_ModelNet40.py", line 283, in
trainer.train(net, training_loader, test_loader, config)
File "/home/ubuntu/src/KPConv-PyTorch/utils/trainer.py", line 188, in train
outputs = net(batch, config)
File "/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, **kwargs)
File "/home/ubuntu/src/KPConv-PyTorch/models/architectures.py", line 143, in forward
x = block_op(x, batch)
File "/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, **kwargs)
File "/home/ubuntu/src/KPConv-PyTorch/models/blocks.py", line 635, in forward
x = self.leaky_relu(self.batch_norm_conv(x))
File "/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, **kwargs)
File "/home/ubuntu/src/KPConv-PyTorch/models/blocks.py", line 456, in forward
x = self.batch_norm(x)
File "/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, **kwargs)
File "/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/nn/modules/batchnorm.py", line 107, in forward
exponential_average_factor, self.eps)
File "/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/nn/functional.py", line 1666, in batch_norm
raise ValueError('Expected more than 1 value per channel when training, got input size {}'.format(size))
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 128, 1])