Commit Graph

11 Commits

Author SHA1 Message Date
Thang Vu
590b96b8aa add setup.py 2022-04-15 08:16: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
11130f850d support distributed eval/test 2022-04-11 07:34:59 +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
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
7d8d7ce10b format with pre-commit 2022-04-08 12:35:12 +00:00
Thang Vu
192f4bdd89 reorganize code 2022-04-08 11:10:38 +00:00