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

Updates and some refactoring

Browse Source
This commit is contained in:
erikwijmans
2017-12-26 19:49:52 -05:00
parent dc4e2b0db3
commit 803d7e1fc6
10 changed files with 211 additions and 226 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/build_ffi.py
+2 -1
View File
@@ -16,7 +16,8 @@ ffi = create_extension(
relative_to=__file__,
with_cuda=True,
extra_objects=extra_objects,
include_dirs=[path.join(base_dir, 'cinclude')])
include_dirs=[path.join(base_dir, 'cinclude')],
verbose=False)
if __name__ == "__main__":
assert torch.cuda.is_available(), "Needs CUDA!"