66 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Thang Vu	3332a478a7	add cuda cast wrapper	2022-04-11 01:17:13 +00:00
Thang Vu	3475ab88b9	support mix precision training	2022-04-10 02:57:11 +00:00
Thang Vu	c620cfc435	support distributed training	2022-04-10 01:36:22 +00:00
Thang Vu	4090322ae0	support spconv2	2022-04-09 04:42:43 +00:00
Thang Vu	a93a1c58bc	make cluster voxelization randomness optional	2022-04-09 03:31:57 +00:00
Thang Vu	068d67b1d4	support evaluate during training	2022-04-09 03:17:01 +00:00
Thang Vu	f811c2f815	minor	2022-04-08 12:40:55 +00:00
Thang Vu	7d8d7ce10b	format with pre-commit	2022-04-08 12:35:12 +00:00
Thang Vu	9a73b2feb1	add pre-commit	2022-04-08 11:58:58 +00:00
Thang Vu	2b2b431fda	relink spconv	2022-04-08 11:14:44 +00:00
Thang Vu	192f4bdd89	reorganize code	2022-04-08 11:10:38 +00:00
Thang Vu	7a20b6d6ae	clean up	2022-04-08 10:25:13 +00:00
Thang Vu	a9e41f3f79	update configs	2022-04-08 10:12:50 +00:00
Thang Vu	0a17ca2cd1	refactor bfs_cluster	2022-04-08 09:49:21 +00:00
Thang Vu	190415587a	format cuda code with clang-format	2022-04-08 06:07:52 +00:00
Thang Vu	97f306635c	refactor train instance	2022-04-08 06:07:27 +00:00
Thang Vu	b5cd952a9e	remove torch.cuda.empty_cache()	2022-04-08 05:53:39 +00:00
Thang Vu	559e870615	Merge remote-tracking branch 'origin/main' into code_refactor	2022-04-08 03:25:13 +00:00
Thang Vu	f7a31c531e	update variable names	2022-04-08 03:19:49 +00:00
Thang Vu	70f2b7454f	refactor logging	2022-04-07 12:49:08 +00:00
Thang Vu	af2c982653	refactor train point wise net	2022-04-07 07:38:38 +00:00
Thang Vu	e5e813ab67	support load checkpoint with different tensor shape	2022-04-06 10:17:10 +00:00
Thang Vu	009cc7c1f0	refactor s3dis dataset	2022-04-06 10:14:29 +00:00
Thang Vu	91c58d1ce3	remove data prefetch	2022-04-06 04:18:47 +00:00
Thang Vu	733b5c48e2	refactor scannetv2	2022-04-05 12:10:20 +00:00
Thang Vu	f822a70dd5	update forward test s3dis	2022-04-05 08:58:56 +00:00
Thang Vu	b62be6b409	refactor forward test	2022-04-04 09:03:32 +00:00
Thang Vu	253ef54373	Update README.md	2022-04-04 18:02:28 +09:00
Thang Vu	79fa20f8bc	Update README.md	2022-03-31 12:54:38 +09:00
Thang Vu	e62bad7aa0	Merge pull request #11 from thangvubk/s3dis_visualization ... add s3dis visualization	2022-03-30 17:10:54 +09:00
Thang Vu	cd13355776	Add visualization guide for both scannet and s3dis	2022-03-30 17:10:27 +09:00
Thang Vu	6eb4d8a77a	Create visualization.md	2022-03-30 17:08:24 +09:00
Thang Vu	e27b3b8e81	add s3dis visualization	2022-03-30 08:08:06 +00:00
Thang Vu	ab147d1358	Update README.md	2022-03-24 13:33:05 +09:00
Thang Vu	40615587f6	fix small bug due to clean up	2022-03-23 08:44:27 +00:00
Thang Vu	5ac64851e8	Update README.md	2022-03-18 13:14:00 +09:00
Thang Vu	d8022d0834	Update README.md	2022-03-12 16:29:02 +09:00
Thang Vu	23858f56f7	add eval det	2022-03-12 07:26:43 +00:00
Thang Vu	1047d41f69	Update README.md	2022-03-12 15:03:57 +09:00
Thang Vu	99ffb9756e	clean up	2022-03-12 05:59:30 +00:00
Thang Vu	a14122aa4b	rename model	2022-03-12 05:51:36 +00:00
Thang Vu	85bb5e8dbd	Update README.md	2022-03-08 10:36:04 +09:00
Thang Vu	6532747902	Update README.md	2022-03-07 20:53:50 +09:00
Thang Vu	714bacd0c9	rename s3dis configs	2022-03-07 11:52:43 +00:00
Thang Vu	168963d20e	Update README.md	2022-03-07 20:50:14 +09:00
Thang Vu	765731e481	Update training details	2022-03-07 20:49:44 +09:00
Thang Vu	b6d155fc56	merge scannet & s3dis training	2022-03-07 11:34:39 +00:00
Thang Vu	d52fb85fdd	Update README.md	2022-03-07 13:52:08 +09:00
Thang Vu	8e08db10ba	Update installation.md	2022-03-07 13:38:28 +09:00
Thang Vu	2c4b642189	Update README.md	2022-03-06 17:17:12 +09:00