-
Notifications
You must be signed in to change notification settings - Fork 639
Closed
Description
when I run train_classification.py
, there is an error:
Traceback (most recent call last):
File "train_classification.py", line 58, in <module>
classifier = PointNetCls(k = num_classes, num_points = opt.num_points)
TypeError: __init__() got an unexpected keyword argument 'num_points'
I noticed that opt.num_points
has been used in
pointnet.pytorch/train_classification.py
Line 40 in 2ec315f
dataset = PartDataset(root = 'shapenetcore_partanno_segmentation_benchmark_v0', classification = True, npoints = opt.num_points) |
so I comment it and run again, but another error:
ValueError: Expected more than 1 value per channel when training, got input size [1, 512]
and this comes from :
Line 47 in 2ec315f
x = F.relu(self.bn4(self.fc1(x))) |
how to solve it? thanks for your help
Metadata
Metadata
Assignees
Labels
No labels