wassname/Pointnet2_PyTorch
1
0
Fork 0
mirror of https://github.com/wassname/Pointnet2_PyTorch.git synced 2026-08-02 12:20:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

allow inputs with 3 channels (use_xyz)

Browse Source
This commit is contained in:
wassname
2018-04-10 13:28:20 +08:00
parent 20af9060dd
commit 0a6b08b83f
6 changed files with 94 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train_cls.py
+1 -1
View File
@@ -115,7 +115,7 @@ if __name__ == "__main__":
tb_log.configure('runs/{}'.format(args.run_name))
model = Pointnet(input_channels=3, num_classes=40)
model = Pointnet(input_channels=3, num_classes=40, use_xyz=False)
model.cuda()
optimizer = optim.Adam(
model.parameters(), lr=args.lr, weight_decay=args.weight_decay