wassname/Pointnet2_PyTorch
1
0
Fork 0
mirror of https://github.com/wassname/Pointnet2_PyTorch.git synced 2026-06-27 16:00:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Some updates

Browse Source
This commit is contained in:
erikwijmans
2018-01-30 16:22:43 -05:00
parent c4ddd6b099
commit 8bce353da4
4 changed files with 15 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/Pointnet2Cls.py
+4 -7
View File
@@ -88,9 +88,8 @@ class Pointnet2MSG(nn.Module):
npoint=512,
radii=[0.1, 0.2, 0.4],
nsamples=[32, 64, 128],
mlps=[[input_channels, 32, 32,
64], [input_channels, 64, 64, 128],
[input_channels, 64, 96, 128]]
mlps=[[input_channels, 64], [input_channels, 128],
[input_channels, 128]]
)
)
@@ -100,9 +99,8 @@ class Pointnet2MSG(nn.Module):
npoint=128,
radii=[0.2, 0.4, 0.8],
nsamples=[16, 32, 64],
mlps=[[input_channels, 64, 64,
128], [input_channels, 128, 128, 256],
[input_channels, 128, 128, 256]]
mlps=[[input_channels, 128], [input_channels, 256],
[input_channels, 256]]
)
)
self.SA_modules.append(
@@ -136,7 +134,6 @@ if __name__ == "__main__":
model = Pointnet2MSG(3)
model.cuda()
optimizer = optim.Adam(model.parameters(), lr=1e-2)
model_fn = model_fn_decorator(nn.CrossEntropyLoss())