105 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
botastic	04358926d0	added ymir datasest preparation	2022-07-17 21:42:01 +02:00
Thang Vu	a87940664d	Update README.md	2022-06-29 06:52:41 +09:00
Thang Vu	ef3d8430f8	Update README.md	2022-06-29 06:26:44 +09:00
Thang Vu	1dd10687af	Update README.md	2022-06-28 15:33:43 +09:00
Thang Vu	3c2b941bf3	Create custom_dataset.md	2022-06-28 15:32:24 +09:00
Thang Vu	737193e984	Delete custom_dataset.md	2022-06-28 15:31:45 +09:00
Thang Vu	8c11f22ae0	Update README.md	2022-06-28 15:27:36 +09:00
Thang Vu	9d85d672d7	Create custom_dataset.md	2022-06-28 15:24:26 +09:00
Thang Vu	fc1dcbca82	Merge pull request #54 from thangvubk/stpls3d ... support stpls3d dataset	2022-06-28 00:43:20 -05:00
Thang Vu	9337a4a69d	minor	2022-06-28 05:40:55 +00:00
Thang Vu	947d8acb13	Merge remote-tracking branch 'origin/main' into stpls3d	2022-06-07 02:08:28 +00:00
Thang Vu	737123a7ee	Create config_explanation.md	2022-06-03 14:25:20 +09:00
Thang Vu	c2a46dc631	add train log	2022-06-03 04:51:44 +00:00
Thang Vu	4dcb1c5eb3	reverse xyz_middle and aug prob	2022-06-03 04:42:24 +00:00
Thang Vu	d8665970f9	fix model.eval() return None in non-distributed mode	2022-05-08 14:24:49 +00:00
Thang Vu	9f0ee8ef1e	update dataset	2022-04-30 13:12:37 +00:00
Thang Vu	365a9d8900	update stpls3d	2022-04-30 12:28:57 +00:00
Thang Vu	fe9b225f40	support stpls3d dataset	2022-04-27 12:24:54 +00:00
Thang Vu	e0c3b4e1ab	Update inference speed	2022-04-27 15:48:34 +09:00
Thang Vu	66ac1587f5	minor - this wont affect output	2022-04-22 06:09:06 +00:00
Thang Vu	f3188a9921	Update README.md	2022-04-16 19:03:35 +09:00
Thang Vu	3fe43242b1	Update traning	2022-04-16 19:02:49 +09:00
Thang Vu	eced35ab1a	Update installation.md	2022-04-16 01:39:09 +09:00
Thang Vu	8f331aad70	Update README.md	2022-04-16 01:31:30 +09:00
Thang Vu	4c1b2876ca	Merge pull request #16 from thangvubk/code_refactor ... Code refactor	2022-04-16 01:22:47 +09:00
Thang Vu	29836f17a1	update license	2022-04-15 16:20:30 +00:00
Thang Vu	4dbfef048b	remove submodule	2022-04-15 16:18:58 +00:00
Thang Vu	096c5e8749	support save results	2022-04-15 16:08:19 +00:00
Thang Vu	590b96b8aa	add setup.py	2022-04-15 08:16:32 +00:00
Thang Vu	3b39cc3bf1	minor	2022-04-15 05:37:01 +00:00
Thang Vu	efe2ddb6b1	separate train val func	2022-04-15 05:31:38 +00:00
Thang Vu	f483165c2a	fix mem leak	2022-04-14 10:49:32 +00:00
Thang Vu	80a663eec6	convert spconv1 to spconv2 checkpoint	2022-04-14 04:10:35 +00:00
Thang Vu	70c86093db	log with distributed parallel	2022-04-13 14:51:38 +00:00
Thang Vu	f0802b75bb	support multiprocess test	2022-04-13 14:49:34 +00:00
Thang Vu	11130f850d	support distributed eval/test	2022-04-11 07:34:59 +00:00
Thang Vu	08eb82816b	minor	2022-04-11 02:03:29 +00:00
Thang Vu	c68be64a35	fix load checkpoint ddp	2022-04-11 02:00:22 +00:00
Thang Vu	40572e2834	add train aug_prob remove val aug	2022-04-11 01:17:44 +00:00
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