steffen/point_to_primitive
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

explicit model argument

Browse Source
This commit is contained in:
Si11ium
2020-06-19 13:35:37 +02:00
parent fe2bc131df
commit 49b373a8a1
3 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
models/point_net_2.py
View File

@@ -23,8 +23,10 @@ class PointNet2(BaseValMixin,
# Dataset
# =============================================================================
self.dataset = self.build_dataset(ShapeNetPartSegDataset, collate_per_segment=True,
npoints=self.params.npoints)
self.dataset = self.build_dataset(ShapeNetPartSegDataset,
collate_per_segment=True,
npoints=self.params.npoints
)
# Model Paramters
# =============================================================================